RuntimeError1 [에러][Pytorch] RuntimeError: Error(s) in loading state_dict 에러:모델 test 과정에서 ./pretrained 폴더의 .pth 파일을 torch.load 하는 부분에서 아래의 오류가 발생했다.model.load_state_dict(checkpoint['model_state_dict']) File "/~/DDAMFN/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1671, in load_state_dict raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(RuntimeError: Error(s) in loading state_dict for DDAMNet: Missing key(s) in state_.. 2024. 7. 21. 이전 1 다음