From 5306348f260cdc5b956a0735db1acbbc04395cb7 Mon Sep 17 00:00:00 2001 From: VectorSL Date: Thu, 15 Oct 2020 16:44:55 +0800 Subject: [PATCH] gpu odelzoo lenet is not suitable for amp --- model_zoo/official/cv/lenet/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model_zoo/official/cv/lenet/train.py b/model_zoo/official/cv/lenet/train.py index 31f6204bf1..73cc3c1dce 100644 --- a/model_zoo/official/cv/lenet/train.py +++ b/model_zoo/official/cv/lenet/train.py @@ -61,7 +61,7 @@ if __name__ == "__main__": keep_checkpoint_max=cfg.keep_checkpoint_max) ckpoint_cb = ModelCheckpoint(prefix="checkpoint_lenet", directory=args.ckpt_path, config=config_ck) - if args.device_target == "CPU": + if args.device_target != "Ascend": model = Model(network, net_loss, net_opt, metrics={"Accuracy": Accuracy()}) else: model = Model(network, net_loss, net_opt, metrics={"Accuracy": Accuracy()}, amp_level="O2")