스택큐힙리스트

PyTorch에서 학습된 모델을 어떻게 저장할 수 있나요? 다음과 같이 읽었습니다.torch.save() / torch.load()은 (Serializable 객체를) 저장하거나 불러오기 위한 것입니다.model.state_dict() / model.load_state_dict()는 모델 상태를 저장하거나 불러오기 위한 것입니다.답변 1그들의 깃허브 저장소에서 this page를 발견했습니다.모델 저장에 대한 권장 접근 방식모델 직렬화 및 복원에는 두 가지 주요 접근 방식이 있습니다.첫 번째(권장)는 모델 매개변수만 저장 및 로드합니다. torch.save(the_model.state_dict(), PATH)그런 다음에: the_model = TheModelClass(*args, **kwargs)the_..