问题
RuntimeError: Error(s) in loading state_dict for ***: Missing key(s) in state_dict:
解决方法
模型先放在GPU上,在加载模型,即
model.cuda()
model.load_state_dict()
参考
https://blog.csdn.net/e01528/article/details/84981058
问题
RuntimeError: Error(s) in loading state_dict for ***: Missing key(s) in state_dict:
解决方法
模型先放在GPU上,在加载模型,即
model.cuda()
model.load_state_dict()
参考
https://blog.csdn.net/e01528/article/details/84981058