diff --git a/train.py b/train.py
index fece2a5..3be2ffc 100644
--- a/train.py
+++ b/train.py
@@ -170,12 +170,12 @@ if __name__ == "__main__":
         model = Model(network,
                       net_loss,
                       net_opt,
-                      metrics={"accuracy": Accuracy()})
+                      metrics={"accuracy"})
     else:
         model = Model(network,
                       net_loss,
                       net_opt,
-                      metrics={"accuracy": Accuracy()},
+                      metrics={"accuracy"},
                       amp_level="O2")
 
     config_ck = CheckpointConfig(