dhwiii's notepad | 딥 러닝, 코덱 일기장

[PyTorch 에러] RuntimeError: Error(s) in loading state_dict for Generator: 본문

■ Working Draft/◎ 에러코드 일기장

[PyTorch 에러] RuntimeError: Error(s) in loading state_dict for Generator:

dhwiii 2021. 6. 16. 22:31

RuntimeError: Error(s) in loading state_dict for Generator:
...

...

...

Unexpected key(s) in state_dict: "epoch", "arch", "best_psnr", "state_dict", "optimizer".

 

아래와 같이 선언되어 있던 부분을

model.load_state_dict(torch.load(args.model_path, map_location=torch.device("cpu")))

아래로 바꾸어주었더니 오류가 해결 되었습니다.

saved_checkpoint = torch.load(args.model_path)
model.load_state_dict(saved_checkpoint, strict=False)

 

저작자표시 비영리 변경금지

'■ Working Draft > ◎ 에러코드 일기장' 카테고리의 다른 글

[Ubuntu Error] TigerVNC설치 후 마우스,키보드 작동 안됨  (0) 2021.07.28
[파이토치 에러] RuntimeError: Freeze_support() Error  (0) 2021.07.27
[Numpy 에러] TypeError: only size-1 arrays can be converted to Python scalars  (0) 2021.05.19
[Tensorflow 에러코드] AttributeError: 'function' object has no attribute 'graph'  (0) 2021.05.18
[Keras 에러코드] ModuleNotFoundError: No module named 'keras_contrib'  (0) 2021.05.17
'■ Working Draft/◎ 에러코드 일기장' Related Articles more
Comments