fix Rednet101 and Swin_large12_384 bs bug for tipc

pull/1572/head
dongshuilong 2021-12-20 07:26:34 +00:00
parent 538ca47eca
commit cf57d54b77
2 changed files with 2 additions and 2 deletions

View File

@ -6,7 +6,7 @@ gpu_list:0|0,1
-o Global.auto_cast:null
-o Global.epochs:lite_train_lite_infer=2|whole_train_whole_infer=120
-o Global.output_dir:./output/
-o DataLoader.Train.sampler.batch_size:8
-o DataLoader.Train.sampler.batch_size:2
-o Global.pretrained_model:null
train_model_name:latest
train_infer_img_dir:./dataset/ILSVRC2012/val

View File

@ -6,7 +6,7 @@ gpu_list:0|0,1
-o Global.auto_cast:null
-o Global.epochs:lite_train_lite_infer=2|whole_train_whole_infer=120
-o Global.output_dir:./output/
-o DataLoader.Train.sampler.batch_size:8
-o DataLoader.Train.sampler.batch_size:2
-o Global.pretrained_model:null
train_model_name:latest
train_infer_img_dir:./dataset/ILSVRC2012/val