OSSInnovation
/
mindspore
1
0
Fork 26
Code Issues Pull Requests 7 Releases Wiki Activity

!12157 fix mobilenetv3 gpu train param 

From: @wuxuejian
Reviewed-by: @liangchenghui,@c_34
Signed-off-by: @liangchenghui
This commit is contained in:
mindspore-ci-bot 2021-02-05 17:14:16 +08:00 committed by Gitee
parent 2d2e824b77 3d185352e0
commit 408159e301
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model_zoo/official/cv/mobilenetv3/train.py
View File

@ -185,7 +185,7 @@ if __name__ == '__main__':
device_target=args_opt.device_target, device_target=args_opt.device_target,
repeat_num=1, repeat_num=1,
batch_size=config_.batch_size, batch_size=config_.batch_size,
run_distribute=False) run_distribute=args_opt.run_distribute)
elif args_opt.device_target == "CPU": elif args_opt.device_target == "CPU":
dataset = create_dataset_cifar(args_opt.dataset_path, dataset = create_dataset_cifar(args_opt.dataset_path,
do_train=True, do_train=True,