Deep-Reinforcement-Learning-Algorithms-with-PyTorch
Deep-Reinforcement-Learning-Algorithms-with-PyTorch copied to clipboard
Published
20 hours ago •
p-christ
p-christ
Implement model saves
Hi. Great work on the library, it's working like a charm.
Right now, only the DQN Agent implements the locally_save_policy that allows for saving the current model. Would it be possible to add this feature on other agents also ?
I think you can implement that by simply saving all the networks of that specific class [not tested], example:
def locally_save_policy(self):
"""Saves the policy"""
torch.save(self.nn_local.state_dict(), "Models/{}_local_network.pt".format(self.agent_name))
torch.save(self.nn_target.state_dict(), "Models/{}_target_network.pt".format(self.agent_name))
I'll have to confirm this.
