30得票1回答
为什么PyTorch被称为PyTorch?

最近我一直在研究深度学习框架,一直在想PyTorch这个名字的起源。 Keras的主页很好地解释了其名称的由来,而像TensorFlow这样的框架,命名背后的道理似乎很清楚。但是对于PyTorch,我似乎找不到它被命名的原因。 当然,我知道“Py-”前缀,也知道PyTorch在某种意义上是To...

13得票3回答
PyTorch 运行时错误:断言“cur_target >= 0 && cur_target < n_classes”失败。

我正在尝试在Pytorch中创建一个基本的二元分类器,用于分类我的玩家在游戏Pong中是在左侧还是右侧。输入是一个1x42x42的图像,标签是我的玩家方向(右侧=1或左侧=2)。 以下是代码:class Net(nn.Module): def __init__(self, input_...

23得票6回答
Lua的科学库?

是否有类似于Scipy的Lua科学包?

11得票1回答
Torch / Lua,如何选择数组或张量的子集?

我正在使用 Torch/Lua,有一个由10个元素组成的数组dataset。dataset = {11,12,13,14,15,16,17,18,19,20} 如果我写dataset[1],我就能读取数组中第一个元素的结构。th&gt; dataset[1] 11 我需要在10个元素中选择...

7得票3回答
将PyTorch代码从CPU迁移到GPU

从https://github.com/spro/practical-pytorch/blob/master/seq2seq-translation/seq2seq-translation.ipynb的教程中可以了解到,有一个USE_CUDA标志,用于控制变量和张量类型在CPU(当为False...

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

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

17得票3回答
Lua中的()()语法有特殊含义吗?

最近我在阅读一些 Lua 源文件时经常遇到这种语法,特别是第二对方括号是什么意思?下面是一个例子,位于第8行的https://github.com/karpathy/char-rnn/blob/master/model/LSTM.lualocal LSTM = {} function LSTM...

12得票6回答
名称“_C”未定义 PyTorch + Jupyter Notebook

我有一些使用pytorch的代码,在我的IDE(pycharm)中可以正常运行。 为了研究,我尝试从jupyter笔记本中运行它。 在笔记本中的代码: from algorithms import Argparser from algorithms import Session def ...

12得票3回答
使用同步均值和方差的多GPU BN层实现方法

我想知道在使用多个GPU进行训练时,实现批量归一化层并同步批量统计的可能方法。 Caffe 或许有一些可以实现的caffe变体,比如这里。但对于BN层,我的理解是它仍然只同步层的输出,而不是均值和方差。也许MPI可以同步均值和方差,但我认为MPI有点难以实现。 Torch 我看到一些评论这...

10得票2回答
PyTorch Network.parameters()缺少一个必需的位置参数:'self'。

在我的主函数中,当我调用PyTorch中的Network.parameters()时,出现了问题: optimizer = optim.SGD(Network.parameters(), lr=0.001, momentum=0.9) 我得到了以下错误代码: TypeError: par...