StopIteration: Caught StopIteration in replica 0 on device 0.解决

原文来自连接
不是第一次遇到了，但是遇到了真不会改，我只用下面两步就好了
1.torch报错：StopIteration: Caught StopIteration in replica 0 on device 0.
原因：多GPU运行此项目报错，可能是torch版本错误。
修改：按照别的博客将 weight = next(self.parameters()).data改为weight = torch.float32

2.仍报错：AttributeError: ‘torch.dtype’ no attribute ‘new’：torch.dtype没有new属性。
原因：因为1出的修改，weight是torch.dtype类，非torch.tensor数据。
修改：于是看源码只是想获取next(self.parameters()).data的数据类型，大部分都是cuda的torch.float32的类型，