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
相關文章
相關標籤/搜索