RuntimeError: Error(s) in loading state_dict for ***: Missing key(s) in state_dict:

缘由,model须要先放到cuda上,根据你的设置,因此是load_state_dict错了。spa 原来:import model = se_resnet20_v2()module model.load_state_dict(torch.load('/home/zj/senetial/save_models/SENet/SEnet_model_2.pth') if use_gpu and len
相关文章
相关标签/搜索