logging when save params and optimizer states (#4534)

apache · Jan 5, 2017 · 1ae2905 · 1ae2905
1 parent ec78752
commit 1ae2905
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/python/mxnet/module/module.py b/python/mxnet/module/module.py
@@ -132,9 +132,13 @@ def save_checkpoint(self, prefix, epoch, save_optimizer_states=False):
             Whether to save optimizer states for continue training
         """
         self._symbol.save('%s-symbol.json'%prefix)
-        self.save_params('%s-%04d.params'%(prefix, epoch))
+        param_name = '%s-%04d.params' % (prefix, epoch)
+        self.save_params(param_name)
+        logging.info('Saved checkpoint to \"%s\"', param_name)
         if save_optimizer_states:
-            self.save_optimizer_states('%s-%04d.states'%(prefix, epoch))
+            state_name = '%s-%04d.states' % (prefix, epoch)
+            self.save_optimizer_states(state_name)
+            logging.info('Saved optimizer state to \"%s\"', state_name)
 
     def _reset_bind(self):
         """Internal function to reset binded state."""