simplify mnist mlp example

dmlc · Oct 22, 2015 · a75a8ef · a75a8ef
1 parent bed3a02
commit a75a8ef
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/examples/mnist/mlp.jl b/examples/mnist/mlp.jl
@@ -31,9 +31,7 @@ train_provider, eval_provider = get_mnist_providers(batch_size)
 estimator = mx.FeedForward(mlp, context=mx.cpu())
 
 # optimizer
-optimizer = mx.SGD(lr_scheduler=mx.FixedLearningRateScheduler(0.1),
-                   mom_scheduler=mx.FixedMomentumScheduler(0.9),
-                   weight_decay=0.00001)
+optimizer = mx.SGD(lr=0.1, momentum=0.9, weight_decay=0.00001)
 
 # fit parameters
 mx.fit(estimator, optimizer, train_provider, epoch_stop=20, eval_data=eval_provider)