Fix Gluon Estimator nightly test (apache#17042)

ptrendx · Dec 11, 2019 · f6fb0ac · f6fb0ac
1 parent 3850284
commit f6fb0ac
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/nightly/estimator/test_estimator_cnn.py b/tests/nightly/estimator/test_estimator_cnn.py
@@ -116,7 +116,7 @@ def test_estimator_cpu():
         # Define estimator
         est = estimator.Estimator(net=net,
                                   loss=loss,
-                                  metrics=mx.metric.Accuracy(),
+                                  train_metrics=mx.metric.Accuracy(),
                                   trainer=trainer,
                                   context=context)
         # Call fit()
@@ -145,7 +145,7 @@ def test_estimator_gpu():
     # Define estimator
     est = estimator.Estimator(net=net,
                               loss=loss,
-                              metrics=acc,
+                              train_metrics=acc,
                               trainer=trainer,
                               context=context)
     # Call fit()

diff --git a/tests/nightly/estimator/test_sentiment_rnn.py b/tests/nightly/estimator/test_sentiment_rnn.py
@@ -197,7 +197,7 @@ def run(net, train_dataloader, test_dataloader, num_epochs, ctx, lr):
     nested_metrics.add([metrics, mx.metric.Accuracy()])
 
     # Define estimator
-    est = estimator.Estimator(net=net, loss=loss, metrics=nested_metrics,
+    est = estimator.Estimator(net=net, loss=loss, train_metrics=nested_metrics,
                               trainer=trainer, context=ctx)
     # Begin training
     est.fit(train_data=train_dataloader, val_data=test_dataloader,