compression: update script

bytedance · Apr 29, 2020 · 89eb7b0 · 89eb7b0
1 parent d1ae98a
commit 89eb7b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/example/mxnet/train_gluon_imagenet_byteps_gc.py b/example/mxnet/train_gluon_imagenet_byteps_gc.py
@@ -402,7 +402,7 @@ def train(ctx):
                     setattr(
                         param, "byteps_compressor_onebit_enable_scale", opt.onebit_scaling)
                 if opt.compress_momentum:
-                    setattr(param, "byteps_momentum_type", "vanilla")
+                    setattr(param, "byteps_momentum_type", "nesterov")
                     setattr(param, "byteps_momentum_mu", opt.momentum)
 
         if opt.compress_momentum: