Commit 21ccbf74 authored by Nicola Gatto's avatar Nicola Gatto
Browse files

Also export parameters with Block

parent 75c338b0
......@@ -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')\
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment