140得票4回答
Pytorch,什么是梯度参数?

我正在阅读PyTorch的文档,发现了一个他们写的例子。 gradients = torch.FloatTensor([0.1, 1.0, 0.0001]) y.backward(gradients) print(x.grad) x是一个初始变量,从中构造了y(一个3维向量)。问题是,g...

93得票6回答
如何检查两个Torch张量或矩阵是否相等?

我需要一个Torch命令,用于检查两个张量是否具有相同的内容,如果它们具有相同的内容则返回TRUE。 例如:local tens_a = torch.Tensor({9,8,7,6}); local tens_b = torch.Tensor({9,8,7,6}); if (tens_a ...

70得票5回答
PyTorch: 如何为自定义数据集使用DataLoaders

如何针对自己的数据(而不仅仅是 torchvision.datasets)使用 torch.utils.data.Dataset 和 torch.utils.data.DataLoader? 是否有一种方法可以使用内置的 DataLoaders 在任何数据集上使用它们用于 TorchVisi...

68得票14回答
升级到CUDA版本后仍然出现“AssertionError:Torch not compiled with CUDA enabled”的错误

我发现这是一个常见问题,但我仍然找不到解决方案。我试图运行一个简单的存储库这里,它使用PyTorch。尽管我刚刚将我的Pytorch升级到了来自pytorch.org的最新CUDA版本(1.2.0),但它仍然会抛出相同的错误。我使用conda和python 3.7,运行在Windows 10上...

65得票3回答
火炬 - 如何更改张量类型?

我创建了从1到3的排列。 th> y = torch.randperm(3 ); th> y 3 2 1 [torch.DoubleTensor of size 3] 现在,我想将y转换为一个Torch.LongTensor。我该怎么做?

57得票1回答
Pytorch。Dataloader中的pin_memory是如何工作的?

我想理解Dataloader中的pin_memory是如何工作的。 根据文档: pin_memory (bool, optional) – If True, the data loader will copy tensors into CUDA pinned memory before r...

49得票2回答
从PyTorch数据集中获取子集

我有一个网络,我想在某些数据集(例如,CIFAR10)上进行训练。我可以通过以下方式创建数据加载器对象: trainset = torchvision.datasets.CIFAR10(root='./data', train=True, ...

49得票3回答
PyTorch和Torch有什么关系?

有两个 PyTorch 代码仓库: https://github.com/hughperkins/pytorch https://github.com/pytorch/pytorch 第一个明确需要 Torch 和 Lua,并且是一个封装,但第二个除了名称之外没有任何关于 Torch 项目...

38得票3回答
Pytorch 运行时错误:参数 #1 'indices' 期望张量的数据类型为 Long,但实际传入了 CUDAType。

我想在我的电脑上重新执行一个GitHub项目进行嵌入式推荐,首先需要将movieLens数据集中的用户和物品进行嵌入,然后使用内积来预测评分,在完成所有组件的集成后,训练时出现了错误。 代码:from lightfm.datasets import fetch_movielens movie...

34得票3回答
TensorFlow中静态计算图和PyTorch中动态计算图有何不同?

在我学习TensorFlow时,其中一个基本概念是计算图,这个图被称为静态的。但是我发现在Pytorch中,这个图被称为动态的。那么TensorFlow中的静态计算图和Pytorch中的动态计算图有什么区别呢?