使用torch.nn.BatchNorm1d出现Tensor for argument #2 ‘weight‘ is on CPU, but expected it to on GPU错误

st_gcn网络定义中使用了torch.nn.BatchNorm1d，在运行网络时出现RuntimeError: Tensor for argument #2 'weight' is on CPU, but expected it to be on GPU (while checking arguments for cudnn_batch_norm)

论坛中的解释是因为在forward函数中直接定义并使用了一个函数，但是源代码中是在__init__函数中定义，forward函数中再使用的

github上好像说是batchnorm1d的通病

最后在__init__函数中将定义的函数转到GPU上得到解决

self.data_bn = nn.BatchNorm1d(in_channels * A.size(1))
self.data_bn.cuda()

使用torch.nn.BatchNorm1d出现Tensor for argument #2 ‘weight‘ is on CPU, but expected it to on GPU错误相关推荐

解决RuntimeError: Tensor for ‘out’ is on CPU, Tensor for argument #1 ‘self’ is on CPU, but expected th
遇到上述错误可能是由于没有将所有变量搬到GPU上去,你可以尝试这样做,并且一定要注意顺序 device = torch.device('cuda' if torch.cuda.is_available ...
Tensor for ‘out‘ is on CPU, Tensor for argument #1 ‘self‘ is on CPU
1.问题模型训练完后进行测试,报错 RuntimeError: Tensor for 'out' is on CPU, Tensor for argument #1 'self' is on CPU ...
Expected tensor for argument #1 'input' to have the same device as tensor for argument #2 'weight';
解决了题目所述问题. 报错完整信息为: RuntimeError: Expected tensor for argument #1 'input' to have the same device as ...
Expected tensor for argument #1 ‘input‘ to have the same device as tensor for argument #2 ‘weight‘；
这种情况出现的原因之一是创建网络的语句没有放进train函数里比如主文件中有train(),test(),eval()......等各个函数,但是创建网络的语句:net=createnet()放在各 ...
torch.nn、(一)
参考 torch.nn.(一) - 云+社区 - 腾讯云目录 torch.nn Parameters Containers Module Sequential ModuleList ModuleDi ...
Pytorch归一化方法讲解与实战：BatchNormalization、LayerNormalization、nn.BatchNorm1d和LayerNorm()和F.normalize()
文章目录 LayerNormalization BatchNormalization F.normalize 这些Normalization的作用都是让数据保持一个比较稳定的分布,从而加速收敛.Bat ...
torch.nn、(二)
参考 torch.nn.(二) - 云+社区 - 腾讯云目录 Recurrent layers RNN LSTM GRU RNNCell LSTMCell GRUCell Transformer l ...
[Pytorch系列-28]：神经网络基础 - torch.nn模块功能列表
作者主页(文火冰糖的硅基工坊):文火冰糖(王文兵)的博客_文火冰糖的硅基工坊_CSDN博客本文网址:https://blog.csdn.net/HiWangWenBing/article/detai ...
nn.BatchNorm1d
本篇博客主要讲解BatchNorm函数的执行过程,需要读者有一定的批归一化的基础,本文例子通俗易懂,如果没有基础也可以阅读在PyTorch中BatchNorm有三个函数,这里主要讲解前两个,后面的就 ...
PyTorch : torch.nn.xxx 和 torch.nn.functional.xxx
PyTorch : torch.nn.xxx 和 torch.nn.functional.xxx 在写 PyTorch 代码时,我们会发现在 torch.nn.xxx 和 torch.nn.funct ...

使用torch.nn.BatchNorm1d出现Tensor for argument #2 ‘weight‘ is on CPU, but expected it to on GPU错误

使用torch.nn.BatchNorm1d出现Tensor for argument #2 ‘weight‘ is on CPU, but expected it to on GPU错误相关推荐

最新文章

热门文章