碰到Pytorch 多GPU TypeError: zip argument #1 must support iteration
这个问题时,考虑如下解决方式:
1.尝试在模型forward的时候,不要输入list类型的数据,而是将其改成tensor的形式。实在不行就分开写。这样可能可以解决问题
2.model 的forward 方法里,输出不能有常量,必须也是要求tensor的格式,先想办法改一下代码,输出常量在多GPU下是行不通的
这个问题时,考虑如下解决方式:
1.尝试在模型forward的时候,不要输入list类型的数据,而是将其改成tensor的形式。实在不行就分开写。这样可能可以解决问题
2.model 的forward 方法里,输出不能有常量,必须也是要求tensor的格式,先想办法改一下代码,输出常量在多GPU下是行不通的