VinF / deer

DEEp Reinforcement learning framework
Other
485 stars 126 forks source link

Refactor model saving/loading #72

Closed taodav closed 5 years ago

taodav commented 5 years ago

Move responsibility of model saving/loading to the LearningAlgo instead of the Agent.

This is to allow model saving/loading through the library the model was written in (ie. torch.save for PyTorch).