728x90 Pytorch1 파이토치 모델 저장 파이토치 모델 저장은 주로 torch.save(model.state_dict(), PATH)를 사용하여 가중치(state_dict)만 저장하는 방식이 권장된다.이 방식은 용량이 작고 유연하며, 추론 시 model.load_state_dict()로 불러온다. 모델 전체를 저장(torch.save(model, PATH))할 수도 있으나, 클래스 구조 의존성 때문에 권장되지 않는다. 1. State Dict 저장 (가중치만)가장 권장되는 방식으로, 파라미터만 저장하므로 모델 클래스 구조가 정의되어 있어야 한다. 저장: torch.save(model.state_dict(), 'model.pth')불러오기:model = TheModelClass() # 모델 클래스 정의 필요model.load_state_dict.. 2026. 4. 30. 이전 1 다음 728x90 반응형