From 7cdbbd09f9ca0cc836897d01fc4cc08ca91a2b1e Mon Sep 17 00:00:00 2001 From: Ye Yuan Date: Mon, 20 Jun 2016 11:58:46 -0700 Subject: [PATCH] save model --- python/mxnet/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/mxnet/model.py b/python/mxnet/model.py index c4dceb3a23b2..6d8204e6f23d 100644 --- 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)