31得票2回答
PyTorch中nn.Linear的类定义是什么?

以下代码中的self.hidden是什么? import torch.nn as nn import torch.nn.functional as F class Network(nn.Module): def __init__(self): super()._...

10得票2回答
如何从PyTorch数据加载器(dataloader)中获取批次迭代的总数?

我有一个问题,就是如何从PyTorch DataLoader中获取批次迭代的总数? 以下是常见的训练代码: for i, batch in enumerate(dataloader): 那么,有没有方法可以获取“for循环”的总迭代次数呢? 在我的NLP问题中,总迭代次数与int(n...

20得票6回答
属性错误:模块'torch'没有属性'_six'。Pytorch中的Bert模型。

我尝试使用pytorch中的BertModel类加载预训练模型。 我在torch下有_six.py,但它仍显示模块'torch'没有属性'_six' import torch from pytorch_pretrained_bert import BertTokenizer, BertMo...

12得票1回答
各种维度输入的高效PyTorch DataLoader collate_fn函数

我在编写PyTorch的DataLoader类的自定义collate_fn函数时遇到了困难。我需要自定义函数是因为我的输入具有不同的维度。 我目前正在尝试编写斯坦福MURA论文的基本实现。该数据集有一组标记研究。一项研究可能包含多个图像。我创建了一个自定义的Dataset类,使用torch....

16得票1回答
获取CUDA_HOME环境路径 PYTORCH

我已经通过Anaconda在我的系统上安装了CUDA,该系统具有2个GPU,Python已经识别出来了。import torch torch.cuda.is_available() true 然而,当我尝试通过它的C API运行一个模型时,我会得到以下错误:~/anaconda3/lib/py...

8得票3回答
错误 libtorch_python.so: 无法打开共享对象文件: 没有那个文件或目录。

我正在尝试实现fastai预训练语言模型,它需要使用torch。运行代码后,我遇到了一些关于导入torch._C的问题。 我在我的linux系统上通过pip安装了python 3.7.1版本的torch 1.0.1.post2和cuda V7.5.17。我得到了以下错误: Tracebac...

7得票1回答
Pytorch中的forward()方法如果有两个以上的输入参数该如何使用?

有人能告诉我forward()方法中多个参数的概念是什么吗? 通常,forward()方法的实现有两个参数: self input 如果一个forward方法具有比这些更多的参数,PyTorch如何使用forward方法。 让我们考虑这个代码库: https://github.com...

16得票1回答
ResNet的state_dict加载中出现了错误:RuntimeError。

我正在使用以下代码加载我的模型。 def load_model(checkpoint_path): ''' Function that loads a checkpoint and rebuilds the model ''' checkpoint = torch.load...

13得票3回答
如何解决PyTorch中由于大小不匹配导致的运行时错误?

我正在尝试使用 PyTorch 实现一个简单的自编码器。我的数据集由 256 x 256 x 3 的图像组成。我已经构建了一个 torch.utils.data.dataloader.DataLoader 对象,其中包含图像存储为张量。当我运行自编码器时,出现了运行时错误: size ...

18得票5回答
如何将一个包含整数的PyTorch张量转换为布尔类型的张量?

我想将一个整数张量转换为布尔值张量。 具体来说,我希望能够有一个函数,将tensor([0,10,0,16])转换为tensor([0,1,0,1]) 在Tensorflow中,这很容易实现,只需使用tf.cast(x, tf.bool)即可。 我希望将所有大于0的整数转换为1,将所有等...