add speedometer

dmlc · Dec 12, 2017 · 5e27ab1 · 5e27ab1
1 parent ebd7404
commit 5e27ab1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git 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
@@ -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()])