From b3867c51d9d823c87bfd40f2963d48eef5f53aca Mon Sep 17 00:00:00 2001 From: wuxuejian Date: Thu, 4 Feb 2021 20:25:58 +0800 Subject: [PATCH] fix mobilenetv3 cpu train param --- model_zoo/official/cv/mobilenetv3/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model_zoo/official/cv/mobilenetv3/train.py b/model_zoo/official/cv/mobilenetv3/train.py index 09724b64db..2e69abc858 100644 --- a/model_zoo/official/cv/mobilenetv3/train.py +++ b/model_zoo/official/cv/mobilenetv3/train.py @@ -214,7 +214,7 @@ if __name__ == '__main__': loss_scale_manager=loss_scale) cb = [Monitor(lr_init=lr.asnumpy())] - if args_opt.run_distribute: + if args_opt.run_distribute and args_opt.device_target != "CPU": ckpt_save_dir = config_gpu.save_checkpoint_path + "ckpt_" + str(get_rank()) + "/" else: ckpt_save_dir = config_gpu.save_checkpoint_path + "ckpt_" + "/"