Hi @chenyuntc, Could you tell me more about using nn.ModuleList? I also get a similar error when I only use CPU.
File “/home/zli/WorkSpace/PyWork/Terraref/panicle_detection/faster_rcnn/network.py”, line 16, in forward
x = self.conv(x)
File “/usr/local/lib/python2.7/dist-packages/torch/nn/modules/module.py”, line 206, in call
result = self.forward(*input, **kwargs)
File “/usr/local/lib/python2.7/dist-packages/torch/nn/modules/conv.py”, line 237, in forward
self.padding, self.dilation, self.groups)
File “/usr/local/lib/python2.7/dist-packages/torch/nn/functional.py”, line 40, in conv2d
return f(input, weight, bias)
RuntimeError: tensors are on different GPUs