更新 'train.py'

get_metrics用法适配最新版本ms2.0
2023-04-13 10:22:30 +08:00 · 2023-04-13 10:22:30 +08:00 · f38ca01dfd
parent fe9d859201
commit f38ca01dfd
1 changed files with 2 additions and 2 deletions
--- 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(