update example

apache · Jul 5, 2018 · 255ef9b · 255ef9b
1 parent 861bdd5
commit 255ef9b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/example/image-classification/common/fit.py b/example/image-classification/common/fit.py
@@ -49,7 +49,8 @@ def _get_lr_scheduler(args, kv):
     steps = [epoch_size * (x - begin_epoch)
              for x in step_epochs if x - begin_epoch > 0]
     if steps:
-        return (lr, mx.lr_scheduler.MultiFactorScheduler(step=steps, factor=args.lr_factor))
+        return (lr, mx.lr_scheduler.MultiFactorScheduler(step=steps, factor=args.lr_factor,
+                                                         base_lr=args.lr))
     else:
         return (lr, None)