diff --git a/test_tipc/configs/bisenetv1/train_infer_python.txt b/test_tipc/configs/bisenetv1/train_infer_python.txt index 0311f411b0..78f35b9e13 100644 --- a/test_tipc/configs/bisenetv1/train_infer_python.txt +++ b/test_tipc/configs/bisenetv1/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:bisenetv1 python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/bisenetv2/train_infer_python.txt b/test_tipc/configs/bisenetv2/train_infer_python.txt index d0baee4fa6..6d1504bc7c 100644 --- a/test_tipc/configs/bisenetv2/train_infer_python.txt +++ b/test_tipc/configs/bisenetv2/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:bisenetv2 python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/ccnet/train_infer_python.txt b/test_tipc/configs/ccnet/train_infer_python.txt index 6f88b0b4c7..b6efd97989 100644 --- a/test_tipc/configs/ccnet/train_infer_python.txt +++ b/test_tipc/configs/ccnet/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:ccnet python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/ddrnet/train_infer_python.txt b/test_tipc/configs/ddrnet/train_infer_python.txt index 7a932522e4..4d36155f5a 100644 --- a/test_tipc/configs/ddrnet/train_infer_python.txt +++ b/test_tipc/configs/ddrnet/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:ddrnet python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu Global.auto_cast:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/deeplabv3p_resnet50/train_infer_python.txt b/test_tipc/configs/deeplabv3p_resnet50/train_infer_python.txt index 1bc6346ef0..c712e200a8 100644 --- a/test_tipc/configs/deeplabv3p_resnet50/train_infer_python.txt +++ b/test_tipc/configs/deeplabv3p_resnet50/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:deeplabv3p_resnet50 python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/deeplabv3p_resnet50_cityscapes/train_infer_python.txt b/test_tipc/configs/deeplabv3p_resnet50_cityscapes/train_infer_python.txt index 0840da538f..7638b8f6cc 100644 --- a/test_tipc/configs/deeplabv3p_resnet50_cityscapes/train_infer_python.txt +++ b/test_tipc/configs/deeplabv3p_resnet50_cityscapes/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:deeplabv3p_resnet50_cityscapes python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/encnet/train_infer_python.txt b/test_tipc/configs/encnet/train_infer_python.txt index b620665703..ec6cc8d26f 100644 --- a/test_tipc/configs/encnet/train_infer_python.txt +++ b/test_tipc/configs/encnet/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:encnet python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/enet/train_infer_python.txt b/test_tipc/configs/enet/train_infer_python.txt index 774eb2b9f0..7c48dcb353 100644 --- a/test_tipc/configs/enet/train_infer_python.txt +++ b/test_tipc/configs/enet/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:enet python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu Global.auto_cast:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/espnetv2/train_infer_python.txt b/test_tipc/configs/espnetv2/train_infer_python.txt index 80e4860eb2..5fa6ec2fb9 100644 --- a/test_tipc/configs/espnetv2/train_infer_python.txt +++ b/test_tipc/configs/espnetv2/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:espnetv2 python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/fastscnn/train_infer_python.txt b/test_tipc/configs/fastscnn/train_infer_python.txt index 67e1e65d39..b7f24018b7 100644 --- a/test_tipc/configs/fastscnn/train_infer_python.txt +++ b/test_tipc/configs/fastscnn/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:fastscnn python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/fcn_hrnetw18/train_infer_python.txt b/test_tipc/configs/fcn_hrnetw18/train_infer_python.txt index c35838b99d..bb88689594 100644 --- a/test_tipc/configs/fcn_hrnetw18/train_infer_python.txt +++ b/test_tipc/configs/fcn_hrnetw18/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:fcn_hrnetw18 python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/fcn_hrnetw18_small/train_infer_python.txt b/test_tipc/configs/fcn_hrnetw18_small/train_infer_python.txt index 046969f28f..03c494e7a6 100644 --- a/test_tipc/configs/fcn_hrnetw18_small/train_infer_python.txt +++ b/test_tipc/configs/fcn_hrnetw18_small/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:fcn_hrnetw18_small python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/fcn_uhrnetw18_small/train_infer_python.txt b/test_tipc/configs/fcn_uhrnetw18_small/train_infer_python.txt index 6833f40286..08dc55215c 100644 --- a/test_tipc/configs/fcn_uhrnetw18_small/train_infer_python.txt +++ b/test_tipc/configs/fcn_uhrnetw18_small/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:fcn_uhrnetw18_small python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/glore/train_infer_python.txt b/test_tipc/configs/glore/train_infer_python.txt index 99c19791e9..be3cd55971 100644 --- a/test_tipc/configs/glore/train_infer_python.txt +++ b/test_tipc/configs/glore/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:glore python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/hrnet_w48_contrast/train_infer_python.txt b/test_tipc/configs/hrnet_w48_contrast/train_infer_python.txt index fc324a2122..631e8c044b 100644 --- a/test_tipc/configs/hrnet_w48_contrast/train_infer_python.txt +++ b/test_tipc/configs/hrnet_w48_contrast/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:hrnet_w48_contrast python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/knet/train_infer_python.txt b/test_tipc/configs/knet/train_infer_python.txt index 01f4e81158..c2ac2fdcd0 100644 --- a/test_tipc/configs/knet/train_infer_python.txt +++ b/test_tipc/configs/knet/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:knet python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: @@ -13,7 +13,7 @@ train_infer_img_dir:test_tipc/data/cityscapes/cityscapes_val_5.list null:null ## trainer:norm_train -norm_train:tools/train.py --config test_tipc/configs/knet/knet_resnet50_cityscapes_1024x512_80k --device gpu --save_interval 500 --seed 100 --num_workers 8 +norm_train:tools/train.py --config test_tipc/configs/knet/knet_resnet50_cityscapes_1024x512_80k.yml --device gpu --save_interval 500 --seed 100 --num_workers 8 pact_train:null fpgm_train:null distill_train:null diff --git a/test_tipc/configs/lpsnet/train_infer_python.txt b/test_tipc/configs/lpsnet/train_infer_python.txt index 752637b4af..af3e463b7a 100644 --- a/test_tipc/configs/lpsnet/train_infer_python.txt +++ b/test_tipc/configs/lpsnet/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:lpsnet python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu Global.auto_cast:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: @@ -13,7 +13,7 @@ train_infer_img_dir:test_tipc/data/cityscapes/cityscapes_val_5.list null:null ## trainer:norm -norm_train:tools/train.py --config test_tipc/configs/lpsnet/lpsnet_m_cityscapes_1536x1024_200k.yml --do_eval --save_interval 500 --seed 100 +norm_train:tools/train.py --config test_tipc/configs/lpsnet/lpsnet_m_cityscapes_1536x1024_200k.yml --do_eval --save_interval 500 --seed 100 --device gpu pact_train:null fpgm_train:null distill_train:null diff --git a/test_tipc/configs/mask2former/train_infer_python.txt b/test_tipc/configs/mask2former/train_infer_python.txt index c88b2dca2e..6dd03aefff 100644 --- a/test_tipc/configs/mask2former/train_infer_python.txt +++ b/test_tipc/configs/mask2former/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:mask2former python:python gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/maskformer/train_infer_python.txt b/test_tipc/configs/maskformer/train_infer_python.txt index aafb16bb2a..74dba4fdde 100644 --- a/test_tipc/configs/maskformer/train_infer_python.txt +++ b/test_tipc/configs/maskformer/train_infer_python.txt @@ -1,8 +1,8 @@ ===========================train_params=========================== model_name:maskformer -python:python3.8 +python:python gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/mobileseg_mv3/train_infer_python.txt b/test_tipc/configs/mobileseg_mv3/train_infer_python.txt index 9ae4acb526..bd83c6fc9f 100644 --- a/test_tipc/configs/mobileseg_mv3/train_infer_python.txt +++ b/test_tipc/configs/mobileseg_mv3/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:mobileseg_mv3 python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/mscale_ocrnet/train_infer_python.txt b/test_tipc/configs/mscale_ocrnet/train_infer_python.txt index 7ca8080f42..6ffb0fa81c 100644 --- a/test_tipc/configs/mscale_ocrnet/train_infer_python.txt +++ b/test_tipc/configs/mscale_ocrnet/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:psa python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/ocrnet_hrformer_base/train_infer_python.txt b/test_tipc/configs/ocrnet_hrformer_base/train_infer_python.txt index fd262c9a1b..94695bc7d3 100644 --- a/test_tipc/configs/ocrnet_hrformer_base/train_infer_python.txt +++ b/test_tipc/configs/ocrnet_hrformer_base/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:ocrnet_hrformer_base python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/ocrnet_hrformer_small/train_infer_python.txt b/test_tipc/configs/ocrnet_hrformer_small/train_infer_python.txt index 3636a7189a..3a9c1a95fe 100644 --- a/test_tipc/configs/ocrnet_hrformer_small/train_infer_python.txt +++ b/test_tipc/configs/ocrnet_hrformer_small/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:ocrnet_hrformer_small python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/ocrnet_hrnetw18/train_infer_python.txt b/test_tipc/configs/ocrnet_hrnetw18/train_infer_python.txt index 3f1515416f..aa4b3584ae 100644 --- a/test_tipc/configs/ocrnet_hrnetw18/train_infer_python.txt +++ b/test_tipc/configs/ocrnet_hrnetw18/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:ocrnet_hrnetw18 python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/ocrnet_hrnetw48/train_infer_python.txt b/test_tipc/configs/ocrnet_hrnetw48/train_infer_python.txt index 5f398f5a74..3ac66eeb44 100644 --- a/test_tipc/configs/ocrnet_hrnetw48/train_infer_python.txt +++ b/test_tipc/configs/ocrnet_hrnetw48/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:ocrnet_hrnetw48 python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/pfpnnet/train_infer_python.txt b/test_tipc/configs/pfpnnet/train_infer_python.txt index b1f6ea0584..8e39fa8124 100644 --- a/test_tipc/configs/pfpnnet/train_infer_python.txt +++ b/test_tipc/configs/pfpnnet/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:pfpnnet python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/pp_liteseg_stdc1/train_infer_python.txt b/test_tipc/configs/pp_liteseg_stdc1/train_infer_python.txt index bd130def3b..371adc6b19 100644 --- a/test_tipc/configs/pp_liteseg_stdc1/train_infer_python.txt +++ b/test_tipc/configs/pp_liteseg_stdc1/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:pp_liteseg_stdc1 python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/pp_liteseg_stdc2/train_infer_python.txt b/test_tipc/configs/pp_liteseg_stdc2/train_infer_python.txt index 1a9e8334d3..4087190d32 100644 --- a/test_tipc/configs/pp_liteseg_stdc2/train_infer_python.txt +++ b/test_tipc/configs/pp_liteseg_stdc2/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:pp_liteseg_stdc2 python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/pphumanseg_lite/train_infer_python.txt b/test_tipc/configs/pphumanseg_lite/train_infer_python.txt index 63d321bcd4..448f35f5a0 100644 --- a/test_tipc/configs/pphumanseg_lite/train_infer_python.txt +++ b/test_tipc/configs/pphumanseg_lite/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:pphumanseg_lite python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu Global.auto_cast:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/pphumanseg_server/train_infer_python.txt b/test_tipc/configs/pphumanseg_server/train_infer_python.txt index 1f2ab40ceb..98721715f9 100644 --- a/test_tipc/configs/pphumanseg_server/train_infer_python.txt +++ b/test_tipc/configs/pphumanseg_server/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:pphumanseg_server python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu Global.auto_cast:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/ppmatting/train_infer_python.txt b/test_tipc/configs/ppmatting/train_infer_python.txt index 803ee1ec68..b27ac1290b 100644 --- a/test_tipc/configs/ppmatting/train_infer_python.txt +++ b/test_tipc/configs/ppmatting/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:ppmatting python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/rtformer/train_infer_python.txt b/test_tipc/configs/rtformer/train_infer_python.txt index 8c4e98449b..e54ce29d0b 100644 --- a/test_tipc/configs/rtformer/train_infer_python.txt +++ b/test_tipc/configs/rtformer/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:rtformer_base python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/segformer_b0/train_infer_python.txt b/test_tipc/configs/segformer_b0/train_infer_python.txt index 206417ca02..dfeab5eab5 100644 --- a/test_tipc/configs/segformer_b0/train_infer_python.txt +++ b/test_tipc/configs/segformer_b0/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:segformer_b0 python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/segnext/train_infer_python.txt b/test_tipc/configs/segnext/train_infer_python.txt index c65bcf3f4a..fa5d50b313 100644 --- a/test_tipc/configs/segnext/train_infer_python.txt +++ b/test_tipc/configs/segnext/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:segnext python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu Global.auto_cast:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: @@ -13,7 +13,7 @@ train_infer_img_dir:test_tipc/data/cityscapes/cityscapes_val_5.list null:null ## trainer:norm -norm_train:tools/train.py --config test_tipc/configs/segnext/segnext_mscan_tiny_1024x1024_cityscapes_160k.yml --do_eval --save_interval 500 --seed 100 +norm_train:tools/train.py --config test_tipc/configs/segnext/segnext_mscan_t_cityscapes_1024x1024_160k.yml --do_eval --save_interval 500 --seed 100 pact_train:null fpgm_train:null distill_train:null diff --git a/test_tipc/configs/sfnet/train_infer_python.txt b/test_tipc/configs/sfnet/train_infer_python.txt index 8732bd089d..f812d7cab9 100644 --- a/test_tipc/configs/sfnet/train_infer_python.txt +++ b/test_tipc/configs/sfnet/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:sfnet python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/stdc_stdc1/train_infer_python.txt b/test_tipc/configs/stdc_stdc1/train_infer_python.txt index 936d990519..9336343500 100644 --- a/test_tipc/configs/stdc_stdc1/train_infer_python.txt +++ b/test_tipc/configs/stdc_stdc1/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:stdc_stdc1 python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/upernet/train_infer_python.txt b/test_tipc/configs/upernet/train_infer_python.txt index 6c1173bb8e..49c3248b53 100644 --- a/test_tipc/configs/upernet/train_infer_python.txt +++ b/test_tipc/configs/upernet/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:upernet python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: diff --git a/test_tipc/configs/vit_adapter/train_infer_python.txt b/test_tipc/configs/vit_adapter/train_infer_python.txt index 5c5de60493..48cbef7dd8 100644 --- a/test_tipc/configs/vit_adapter/train_infer_python.txt +++ b/test_tipc/configs/vit_adapter/train_infer_python.txt @@ -2,7 +2,7 @@ model_name:vit_adapter python:python3.7 gpu_list:0|0,1 -Global.use_gpu:null|null +--device:gpu --precision:null --iters:lite_train_lite_infer=20|lite_train_whole_infer=20|whole_train_whole_infer=1000 --save_dir: