오영제 선생님 강의 잘 듣고 있고 감사드립니다.드리고 싶은 질문은 강화학습은 학습 후 훈련내용을 어떻게 저장하고 필요시 어떻게 로드하는 지 궁금하니다.

강화학습 모델은 일반 딥러닝 모델보다 저장할 내용이 많습니다. 간단히 요약하면,<ul><li>모델은 <code>state_dict()</code></li><li>옵티마이저도 <code>state_dict()</code></li><li>episode, step, epsilon 같은 값은 숫자 그대로 저장</li><li>정규화 통계는 dict로 저장</li><li>replay buffer는 직렬화 가능한 형태로 변환해서 저장</li><li>이 모든 것을 하나의 checkpoint dict에 넣고 <code>torch.save()</code>로 저장</li></ul>구체적인 코드는 다음과 같습니다.<a target="_blank" rel="noopener noreferrer nofollow" href="http://torch.save">torch.save</a>({ "model_state_dict": ..., "optimizer_state_dict": ..., "train_state": ..., "buffer": ..., "config": ...,}, "checkpoint.pth")감사합니다.

Inflearn コミュニティ Q&A

강화학습저장 및 로드