This is an automated email from the ASF dual-hosted git repository.
zhasheng pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/master by this push:
new 83ecf2d Adding missing optimizers that use momentum. (#11826)
83ecf2d is described below
commit 83ecf2da4bf9691589d9818ad662c5eae09601c5
Author: vishaalkapoor <[email protected]>
AuthorDate: Thu Jul 19 20:00:14 2018 -0700
Adding missing optimizers that use momentum. (#11826)
This addresses issue 11620.
---
example/image-classification/common/fit.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/example/image-classification/common/fit.py
b/example/image-classification/common/fit.py
index ca1c09a..67cda78 100755
--- a/example/image-classification/common/fit.py
+++ b/example/image-classification/common/fit.py
@@ -216,7 +216,7 @@ def fit(args, network, data_loader, **kwargs):
'multi_precision': True}
# Only a limited number of optimizers have 'momentum' property
- has_momentum = {'sgd', 'dcasgd', 'nag'}
+ has_momentum = {'sgd', 'dcasgd', 'nag', 'signum', 'lbsgd'}
if args.optimizer in has_momentum:
optimizer_params['momentum'] = args.mom
