From 3d185352e069fb437f8e1ca6b0064d9e540f5b4e Mon Sep 17 00:00:00 2001 From: wuxuejian Date: Fri, 5 Feb 2021 10:44:10 +0800 Subject: [PATCH] fix mobilenetv3 gpu 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 2e69abc8581..8d07a8e9bb8 100644 --- a/model_zoo/official/cv/mobilenetv3/train.py +++ b/model_zoo/official/cv/mobilenetv3/train.py @@ -185,7 +185,7 @@ if __name__ == '__main__': device_target=args_opt.device_target, repeat_num=1, batch_size=config_.batch_size, - run_distribute=False) + run_distribute=args_opt.run_distribute) elif args_opt.device_target == "CPU": dataset = create_dataset_cifar(args_opt.dataset_path, do_train=True,