标签:pytorch
-
在多个GPU上运行LSTM可获得“输入张量和隐藏张量不在同一设备上” - python
我正在尝试在pytorch中训练LSTM层。我正在使用4个GPU。初始化时,我添加了.cuda()函数将隐藏层移动到GPU。但是,当我使用多个GPU运行代码时,出现此运行时错误: RuntimeError: Input and hidden tensors are not at the same device 我试图通过在正向函数中使用.cuda()函数来解 […]
-
pytorch:我们可以直接在forward()函数中使用nn.Module层吗? - python
通常,在构造函数中,我们声明要使用的所有层。在前向功能中,我们定义了如何从输入到输出运行模型。 我的问题是,如果直接在forward()函数中调用那些预定义/内置的 nn。这种Keras function API样式适用于 Pytorch 吗?如果没有,为什么? 更新:以这种方式构造的TestModel 确实运行成功,没有发出警报。但是与传统方法相比,训练损 […]
-
PyTorch和TensorFlow对象检测-评估-类型<class'numpy.float64'>的对象不能安全地解释为整数 - python
我正在尝试运行以下PyTorch人检测示例: https://pytorch.org/tutorials/intermediate/torchvision_tutorial.html 我正在使用Ubuntu 18.04。以下是我已执行的步骤的摘要: 1)在具有GTX 1650 GPU的Lenovo ThinkPad X1 Extreme Gen 2上安装了U […]
-
需要帮助了解pytorch中的梯度函数 - python
以下代码 w = np.array([[2., 2.],[2., 2.]]) x = np.array([[3., 3.],[3., 3.]]) b = np.array([[4., 4.],[4., 4.]]) w = torch.tensor(w, requires_grad=True) x = torch.tensor(x, requires_grad […]
-
PyTorch-将张量与标量相乘会得出零向量 - python
我不知道为什么张量结果全为0。这里有什么问题吗? >>> import torch >>> import numpy as np >>> import math >>> torch.__version__ '0.4.1' >>> np.__versi […]
-
RuntimeError:cuda运行时错误(30):火炬/csrc/cuda/Module.cpp处的未知错误:32 - python
using config: 'CONFIG_NAME': 'stageI', 'CUDA': True, 'DATASET_NAME': 'coco', 'DATA_DIR': '/home/anwar/Documents/Sta […]
-
如何使用双向RNN和pytorch填补空白? - python
我正在尝试使用双向RNN和pytorch填补空白。 输入如下:The dog is _____, but we are happy he is okay. 输出将如下所示: 1. hyper (Perplexity score here) 2. sad (Perplexity score here) 3. scared (Perplexity score h […]
-
使用Pytorch进行深度学习:了解神经网络示例 - python
我正在阅读Pytorch documentation,并且对引入的神经网络有几个问题。该文档定义了以下网络: import torch import torch.nn as nn import torch.nn.functional as F class Net(nn.Module): def __init__(self): super(Net, self) […]
-
如何修改此PyTorch卷积神经网络以接受64 x 64图像并正确输出预测? - python
我从here获得了这个卷积神经网络(CNN)。它接受32 x 32图像,默认为10类。但是,我有500个班级的64 x 64图像。当我传递64 x 64图像(批量大小保持恒定为32)时,出现以下错误。 ValueError: Expected input batch_size (128) to match target batch_size (32). Th […]
-
我的PyTorch前进功能可以执行其他操作吗? - python
通常,forward函数将一堆图层串在一起,并返回最后一层的输出。在返回之前,我可以在最后一层之后做一些其他处理吗?例如,一些标量乘法和通过.view重塑? 我知道autograd会以某种方式找出梯度。因此,我不知道我的额外处理是否会导致这种情况。谢谢。 参考方案 pytorch通过张量的computational graph而不是通过函数跟踪渐变。只要您的 […]