Aufgrund einer Wartung wird GitLab am 21.09. zwischen 8:00 und 9:00 Uhr kurzzeitig nicht zur Verfügung stehen. / Due to maintenance, GitLab will be temporarily unavailable on 21.09. between 8:00 and 9:00 am.

Commit 7a8c3ae8 authored by Nicola Gatto's avatar Nicola Gatto
Browse files

Export parameters after save_parameter is called

Additionally use the export method of gluon to use the parameters with
the C++ predictor
parent 053bf612
...@@ -270,6 +270,7 @@ class Agent(object): ...@@ -270,6 +270,7 @@ class Agent(object):
filedir = self._output_directory if filedir is None else filedir filedir = self._output_directory if filedir is None else filedir
filename = os.path.join(filedir, filename + '.params') filename = os.path.join(filedir, filename + '.params')
net.save_parameters(filename) net.save_parameters(filename)
net.export(filename, epoch=0)
def save_best_network(self, path, epoch=0): def save_best_network(self, path, epoch=0):
self._logger.info( self._logger.info(
......
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