VQA-ReGat:关系感知图形注意网络用于VQA
-
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的类型,
因此最终修改:
weight = 0
weight = torch.tensor

本文针对VQA-ReGat项目中遇到的多GPU运行错误进行了详细解析,并提供了有效的解决方案,包括对torch数据类型的调整及内存管理优化等。
最低0.47元/天 解锁文章
1343

被折叠的 条评论
为什么被折叠?



