diff --git a/examples/mnist/lenet-stn.jl b/examples/mnist/lenet-stn.jl index 60f2def68..1720bf780 100644 --- a/examples/mnist/lenet-stn.jl +++ b/examples/mnist/lenet-stn.jl @@ -66,4 +66,5 @@ initializer = mx.XavierInitializer(distribution=mx.xv_normal, mx.fit(model, optimizer, train_provider, n_epoch=20, eval_data=eval_provider, - initializer=initializer) + initializer=initializer, + callbacks=[mx.speedometer()]) diff --git a/examples/mnist/lenet.jl b/examples/mnist/lenet.jl index af3e8c41d..3fa4e4249 100644 --- a/examples/mnist/lenet.jl +++ b/examples/mnist/lenet.jl @@ -42,4 +42,7 @@ model = mx.FeedForward(lenet, context=mx.gpu()) optimizer = mx.SGD(lr=0.05, momentum=0.9, weight_decay=0.00001) # fit parameters -mx.fit(model, optimizer, train_provider, n_epoch=20, eval_data=eval_provider) +mx.fit(model, optimizer, train_provider, + n_epoch=20, + eval_data=eval_provider, + callbacks=[mx.speedometer()])