모델 저장을 잊지 마

1. 모델 전체 저장

torch.save(model, PATH)

# 모델 클래스는 어딘가에 반드시 선언되어 있어야 합니다.
model = torch.load(PATH)
model.eval()

2. 학습 가능한 매개변수(weight, bias)가 담겨있는 딕셔너리만 저장:

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

model = TheModelClass(*args, **kwargs)
model.load_state_dict(torch.load(PATH))
model.eval()

https://tutorials.pytorch.kr/beginner/saving_loading_models.html

인간은 왜 같은 실수를 반복할까?