Skip to content
Snippets Groups Projects
Commit 21ccbf74 authored by Nicola Gatto's avatar Nicola Gatto
Browse files

Also export parameters with Block

parent 75c338b0
Branches
No related tags found
3 merge requests!20Implemented layer variables and RNN layer,!19Integrate TD3 Algorithm and Gaussian Noise,!18Integrate TD3 Algorithm and Gaussian Noise
......@@ -269,6 +269,7 @@ class Agent(object):
filename = filename + '-ep{}'.format(episode)
net.export(filename, epoch=0)
net.save_parameters(filename + '.params')
def export_best_network(self, path=None, epoch=0):
path = os.path.join(self._output_directory, 'best_network')\
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment