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):
filedir = self._output_directory if filedir is None else filedir
filename = os.path.join(filedir, filename + '.params')
net.export(filename, epoch=0)
def save_best_network(self, path, epoch=0):
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