add speedometer

Author: iblislin

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()])

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()])