codeloversky issues

Repositories
Issues
Comments

Results 1 issues of


                                            codeloversky

AttributeError: 'dict' object has no attribute 'state_dict'

你好，这个模型跑起来还是挺强的，我用cpu模型可以正常使用，但是在换成gpu模式的时候出了点问题，在使用n卡模型运行时提示出错Loading model for gpu... Traceback (most recent call last): File "run.py", line 90, in m2 = torch.load(MODEL_NAME + '.pth', map_location='cpu').state_dict() AttributeError: 'dict' object has no attribute 'state_dict' 都检查了一遍，模型没问题，网上查找了很多相关的问题，都没能解决，能否请教一下怎么解决掉？非常感谢！！