Aufgrund einer Störung des s3 Storage, könnten in nächster Zeit folgende GitLab Funktionen nicht zur Verfügung stehen: LFS, Container Registry, Job Artifacs, Uploads (Wiki, Bilder, Projekt-Exporte). Wir bitten um Verständnis. Es wird mit Hochdruck an der Behebung des Problems gearbeitet. Weitere Informationen zur Störung des Object Storage finden Sie hier: https://maintenance.itc.rwth-aachen.de/ticket/status/messages/59-object-storage-pilot

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.save_parameters(filename)
net.export(filename, epoch=0)
def save_best_network(self, path, epoch=0):
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