You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
##报错的提示:
RuntimeError: Expected tensor for argument #1 'input' to have the same device as tensor for argument #2 'weight'; but device 0 does not equal 1 (while checking arguments for cudnn_convolution)
期待您的帮助,我将不胜感激。
The text was updated successfully, but these errors were encountered:
我有两块GPU 0和1,除了在class Opion() 类中指定 ### 参数 self.gpu_ids=[1] 外,代码中还需要在哪块有指定,我只改了这个它显示了以下错误:(### train.ipy运行到 model.optimize_parameters()出错)。
调用CSA.py 中的224行出现报错
223 def optimize_parameters(self):
--> 224 self.forward()
##报错的提示:
RuntimeError: Expected tensor for argument #1 'input' to have the same device as tensor for argument #2 'weight'; but device 0 does not equal 1 (while checking arguments for cudnn_convolution)
期待您的帮助,我将不胜感激。
The text was updated successfully, but these errors were encountered: