diff --git a/tests/nightly/estimator/test_estimator_cnn.py b/tests/nightly/estimator/test_estimator_cnn.py index 4a3bb2076524..af519536dbed 100644 --- 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 index 233355b7ebfd..ab124ba95db3 100644 --- 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,