VQA-ReGat 项目运行遇到的错误

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

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
评论 19
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值