pytorch
czg792845236
这个作者很懒,什么都没留下…
展开
-
pytorch scatter_的用法及含义
待添加原创 2020-04-18 19:32:27 · 3827 阅读 · 1 评论 -
pytorch损失函数binary_cross_entropy和binary_cross_entropy_with_logits的区别
binary_cross_entropy和binary_cross_entropy_with_logits都是来自torch.nn.functional的函数,首先对比官方文档对它们的区别:函数名解释binary_cross_entropyFunction that measures the Binary Cross Entropy between the target a...原创 2020-04-18 15:58:45 · 55055 阅读 · 4 评论 -
Pytorch Dataset,TensorDataset,Dataloader,Sampler关系
DataloaderDataloader是数据加载器,组合数据集和采样器,并在数据集上提供单线程或多线程的迭代器。所以Dataloader的参数必然需要指定数据集Dataset和采样器Sampler。class torch.utils.data.DataLoader(dataset, batch_size=1, shuffle=False, sampler=None, num_workers...原创 2020-04-09 19:22:56 · 1007 阅读 · 0 评论