에러(Error) 이야기2 [Pytorch] RuntimeError: Error(s) in loading state_dict for RobertaForSequenceClassification 1. 에러 발생 상황 학습된 모델을 저장하고 다시 불러오는(model.load_state_dict()) 상황에서 발생하였다.. 에러 메시지는 다음과 같다. 2. 해결 방법 찾아보니 버전차이가 원인이였다. fine-tuning을 진행한 pytorch의 버전은 2.2버전이고, inference하기 위해 불러온 환경에서의 pytorch 버전은 1.7 버전이였다. 아마 버전차이로 인해 에러가 발생하는것 같다. 해결방법은 의외로 간단하다. load_state_dict의 인자 중 strict를 False로 주면 바로 해결된다.(load할 수 있는 key값만 불러오는 방식인듯) model.load_state_dict(torch.load("./model/saved_model.pt"),strict = False) 위와 .. 2024. 4. 9. [VScode-SSH] failed to parse remote port from server output 평소에 SSH를 이용하여 원격으로 밖에서 VScode를 주로 사용하고 있었는데 갑자기 어느 날부터 ssh 연결이 안 됐다. 아무것도 안 했는데 갑자기 ssh 연결이 안 되니까 원인이 무엇인지도 파악하기 힘들었기 때문에 이 error를 해결하기 참 난감했다. 필자는 노트북으로 M1 mac book air를 사용하고 있었고, 집에 있는 데스크톱은 Windows를 사용하고 있었다. 우선 정말 ssh 접속 자체가 안되는지 알아보기 위해 터미널로 ssh 연결을 시도하였고, 다행히(?) 너무나도 잘 연결되고 있었다. 위 상황을 보면 ssh 연결 자체는 문제가 없다는 뜻이기에 vscode 자체의 문제라고 파악하였다. 우선 내가 본 에러는 다음과 같았다.(캡처는 고칠 당시에 생각도 못했다..) failed to par.. 2022. 11. 16. 이전 1 다음
