fix some regression errors

apache · Dec 9, 2019 · a1b95b9 · a1b95b9
1 parent ebd8c67
commit a1b95b9
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/python/unittest/test_gluon_event_handler.py b/tests/python/unittest/test_gluon_event_handler.py
@@ -219,10 +219,10 @@ def test_logging_interval():
  num_epochs = 1
  ce_loss = loss.SoftmaxCrossEntropyLoss()
  acc = mx.metric.Accuracy()
- logging = LoggingHandler(train_metrics=[acc], log_interval=log_interval)
+ logging = LoggingHandler(metrics=[acc], log_interval=log_interval)
  est = estimator.Estimator(net=net,
  loss=ce_loss,
- metrics=acc)
+ train_metrics=acc)
 
  est.fit(train_data=dataloader,
  epochs=num_epochs,
@@ -244,10 +244,10 @@ def test_logging_interval():
  sys.stdout = mystdout = StringIO()
  acc = mx.metric.Accuracy()
  log_interval = 5
- logging = LoggingHandler(train_metrics=[acc], log_interval=log_interval)
+ logging = LoggingHandler(metrics=[acc], log_interval=log_interval)
  est = estimator.Estimator(net=net,
  loss=ce_loss,
- metrics=acc)
+ train_metrics=acc)
  est.fit(train_data=dataloader,
  epochs=num_epochs,
  event_handlers=[logging])