From 098108a6e1240c4cf37f1786e97b474dca7b8c70 Mon Sep 17 00:00:00 2001
From: Nicola Gatto <nicola.gatto@rwth-aachen.de>
Date: Wed, 17 Jul 2019 17:02:13 +0200
Subject: [PATCH] Only use export method of network for saving parameters
---
src/main/resources/templates/gluon/reinforcement/agent/Agent.ftl | 1 -
1 file changed, 1 deletion(-)
diff --git a/src/main/resources/templates/gluon/reinforcement/agent/Agent.ftl b/src/main/resources/templates/gluon/reinforcement/agent/Agent.ftl
index 58d98edc..8ac5684e 100644
--- a/src/main/resources/templates/gluon/reinforcement/agent/Agent.ftl
+++ b/src/main/resources/templates/gluon/reinforcement/agent/Agent.ftl
@@ -272,7 +272,6 @@ class Agent(object):
else:
self._logger.debug('Saving model parameters')
- net.save_parameters(filename + '.params')
net.export(filename, epoch=0)
def export_best_network(self, path=None, epoch=0):
--
GitLab