Trouble Shooting

Pycharm에서 경로를 못 찾을 경우

- 가상환경에서 OpenCV를 설치하였으나, 경로 읽어오지 못할 경우

- e.g. C:\Users\jhon\anaconda3\envs\torch\Library\bin

- Console > Python Console > 환경 변수에 직접 추가해 주면 된다.

- 파이참 2020버전에서는 문제가 발생하진 않지만, 다른 컴퓨터 2018버전에서는 이런 문제가 발생한다.

 

RuntimeError: freeze_support() Error 해결 방법

aigong.tistory.com/136

 

torch.load(model)에러

모델 저장시에  torch.save(model) 로 모델 아키텍쳐 전체를 저장하는 것이 아니라,

가중치 만을 저장하는 것을 권장한다.

torch.save(model.state_dict(), PATH)

 

만약에 모델 아키텍쳐 전체를 저장했다면,

torch.load(PATH)로 불러올 수 있다.

 

정리하면,

# 1) 과거 방식 - 모델 전체를 저장했을 시 불러오는 방법
model = torch.load(PATH + '\\' + model)  # old
model.eval()
# 혹은 
model = EfficientNet.from_pretrained('efficientnet-b7')
model.load_state_dict(torch.load(PATH + '\\' + model)['state_dict'])

# 2) 현재 방식 - 모델에서 가중치만 저장 했기 때문에 가중치만 읽어온다.
model_bone = EfficientNet.from_pretrained('efficientnet-b7')
model_bone.load_state_dict(torch.load(PATH +'\\'+ model))
model_bone.eval()

 

discuss.pytorch.org/t/error-while-loading-state-dict/49086/2

stackoverflow.com/questions/61242966/pytorch-attributeerror-function-object-has-no-attribute-copy