compression: update script

jasperzhong · jasperzhong · commit a77a9d6265e8 · 2020-04-29T15:49:30.000+08:00
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:
