maodong2056

多GPU训练问题

4

File "scripts/whiteboxgan.py", line 186, in training_step vgg_output = self.pretrained(output) File "/usr/local/python3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 532, in __call__ result = self.forward(*input, **kwargs) File "/home/cyr/17_style/networks/pretrainnet.py", line 172, in forward return self._forward_impl(x) File "/home/cyr/17_style/networks/pretrainnet.py", line...