save model

ijkguo · Jun 20, 2016 · 7cdbbd0 · 7cdbbd0
1 parent d2eb9d0
commit 7cdbbd0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/mxnet/model.py b/python/mxnet/model.py
@@ -339,7 +339,7 @@ def save_checkpoint(prefix, epoch, symbol, arg_params, aux_params):
     - ``prefix-symbol.json`` will be saved for symbol.
     - ``prefix-epoch.params`` will be saved for parameters.
     """
-    #symbol.save('%s-symbol.json' % prefix)
+    symbol.save('%s-symbol.json' % prefix)
     save_dict = {('arg:%s' % k) : v for k, v in arg_params.items()}
     save_dict.update({('aux:%s' % k) : v for k, v in aux_params.items()})
     param_name = '%s-%04d.params' % (prefix, epoch)