zhangweiweicpp

我有两块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...