example: fix MSE init (#376)

dmlc · Dec 15, 2017 · 881759f · 881759f
1 parent 6609616
commit 881759f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/regression-example.jl b/examples/regression-example.jl
@@ -63,15 +63,15 @@ optimizer = mx.ADAM()
 trainprovider, evalprovider = data_source(#= batchsize =# 200)
 mx.fit(model, optimizer, trainprovider,
        initializer = mx.NormalInitializer(0.0, 0.1),
-       eval_metric = mx.MSE{mx.NDArray{Float32,1}}(),
+       eval_metric = mx.MSE(),
        eval_data = evalprovider,
        n_epoch = 20,
        callbacks = [mx.speedometer()])
 # more training with the full sample
 trainprovider, evalprovider = data_source(#= batchsize =# samplesize)
 mx.fit(model, optimizer, trainprovider,
        initializer = mx.NormalInitializer(0.0, 0.1),
-       eval_metric = mx.MSE{mx.NDArray{Float32,1}}(),
+       eval_metric = mx.MSE(),
        eval_data = evalprovider,
        n_epoch = 500,  # previous setting is batchsize = 200, epoch = 20
                        # implies we did (5000 / 200) * 20 times update in previous `fit`