diff --git a/test_tipc/static/ResNet50/N1C1/ResNet50_bs128_amp_fp16_SingleP_DP.sh b/test_tipc/static/ResNet50/N1C1/ResNet50_bs128_amp_fp16_SingleP_DP.sh new file mode 100644 index 0000000000000000000000000000000000000000..0dfc2ceafa9e8a8b446f49ff750c15b15ce8d333 --- /dev/null +++ b/test_tipc/static/ResNet50/N1C1/ResNet50_bs128_amp_fp16_SingleP_DP.sh @@ -0,0 +1,13 @@ +model_item=ResNet50 +bs_item=128 +fp_item=amp_fp16 +run_process_type=SingleP +run_mode=DP +device_num=N1C1 +max_epochs=1 +num_workers=8 + +# get data +bash test_tipc/static/${model_item}/benchmark_common/prepare.sh +# run +bash test_tipc/static/${model_item}/benchmark_common/run_benchmark.sh ${model_item} ${bs_item} ${fp_item} ${run_process_type} ${run_mode} ${device_num} ${max_epochs} ${num_workers} 2>&1; diff --git a/test_tipc/static/ResNet50/N1C1/ResNet50_bs128_fp32_SingleP_DP.sh b/test_tipc/static/ResNet50/N1C1/ResNet50_bs128_fp32_SingleP_DP.sh new file mode 100644 index 0000000000000000000000000000000000000000..0701cdef7ddca07a87f552109b83120d88b98fea --- /dev/null +++ b/test_tipc/static/ResNet50/N1C1/ResNet50_bs128_fp32_SingleP_DP.sh @@ -0,0 +1,17 @@ +model_item=ResNet50 +bs_item=128 +fp_item=fp32 +run_process_type=SingleP +run_mode=DP +device_num=N1C1 +max_epochs=1 +num_workers=8 + +# get data +bash test_tipc/static/${model_item}/benchmark_common/prepare.sh +# run +bash test_tipc/static/${model_item}/benchmark_common/run_benchmark.sh ${model_item} ${bs_item} ${fp_item} ${run_process_type} ${run_mode} ${device_num} ${max_epochs} ${num_workers} 2>&1; +# run profiling +sleep 10; +export PROFILING=true +bash test_tipc/static/${model_item}/benchmark_common/run_benchmark.sh ${model_item} ${bs_item} ${fp_item} ${run_process_type} ${run_mode} ${device_num} ${max_epochs} ${num_workers} 2>&1; diff --git a/test_tipc/static/ResNet50/N1C1/ResNet50_bs128_pure_fp16_SingleP_DP.sh b/test_tipc/static/ResNet50/N1C1/ResNet50_bs128_pure_fp16_SingleP_DP.sh new file mode 100644 index 0000000000000000000000000000000000000000..7a9e82c0e7729006ec88642da79d0a541875dd5e --- /dev/null +++ b/test_tipc/static/ResNet50/N1C1/ResNet50_bs128_pure_fp16_SingleP_DP.sh @@ -0,0 +1,13 @@ +model_item=ResNet50 +bs_item=128 +fp_item=pure_fp16 +run_process_type=SingleP +run_mode=DP +device_num=N1C1 +max_epochs=1 +num_workers=8 + +# get data +bash test_tipc/static/${model_item}/benchmark_common/prepare.sh +# run +bash test_tipc/static/${model_item}/benchmark_common/run_benchmark.sh ${model_item} ${bs_item} ${fp_item} ${run_process_type} ${run_mode} ${device_num} ${max_epochs} ${num_workers} 2>&1; diff --git a/test_tipc/static/ResNet50/N1C1/ResNet50_bs256_fp16_SingleP_DP.sh b/test_tipc/static/ResNet50/N1C1/ResNet50_bs256_amp_fp16_SingleP_DP.sh similarity index 95% rename from test_tipc/static/ResNet50/N1C1/ResNet50_bs256_fp16_SingleP_DP.sh rename to test_tipc/static/ResNet50/N1C1/ResNet50_bs256_amp_fp16_SingleP_DP.sh index b338608ba31f612bb661706a8ab9e17dcd51f5d6..a98ec94fed193feca91d05f9ba468a7c268b8913 100644 --- a/test_tipc/static/ResNet50/N1C1/ResNet50_bs256_fp16_SingleP_DP.sh +++ b/test_tipc/static/ResNet50/N1C1/ResNet50_bs256_amp_fp16_SingleP_DP.sh @@ -1,6 +1,6 @@ model_item=ResNet50 bs_item=256 -fp_item=fp16 +fp_item=amp_fp16 run_process_type=SingleP run_mode=DP device_num=N1C1 diff --git a/test_tipc/static/ResNet50/N1C1/ResNet50_bs256_pure_fp16_SingleP_DP.sh b/test_tipc/static/ResNet50/N1C1/ResNet50_bs256_pure_fp16_SingleP_DP.sh new file mode 100644 index 0000000000000000000000000000000000000000..554455e61e620ca2295faed16f7c4fa6d965ee6e --- /dev/null +++ b/test_tipc/static/ResNet50/N1C1/ResNet50_bs256_pure_fp16_SingleP_DP.sh @@ -0,0 +1,13 @@ +model_item=ResNet50 +bs_item=256 +fp_item=pure_fp16 +run_process_type=SingleP +run_mode=DP +device_num=N1C1 +max_epochs=1 +num_workers=8 + +# get data +bash test_tipc/static/${model_item}/benchmark_common/prepare.sh +# run +bash test_tipc/static/${model_item}/benchmark_common/run_benchmark.sh ${model_item} ${bs_item} ${fp_item} ${run_process_type} ${run_mode} ${device_num} ${max_epochs} ${num_workers} 2>&1; diff --git a/test_tipc/static/ResNet50/N1C1/ResNet50_bs64_fp16_SingleP_DP.sh b/test_tipc/static/ResNet50/N1C1/ResNet50_bs64_amp_fp16_SingleP_DP.sh similarity index 95% rename from test_tipc/static/ResNet50/N1C1/ResNet50_bs64_fp16_SingleP_DP.sh rename to test_tipc/static/ResNet50/N1C1/ResNet50_bs64_amp_fp16_SingleP_DP.sh index c6d96b1f84b029fddd903c8ef6cbeaa22fb7a23d..4b86a74ad3006c2ee462117fd7c959850c0192b2 100644 --- a/test_tipc/static/ResNet50/N1C1/ResNet50_bs64_fp16_SingleP_DP.sh +++ b/test_tipc/static/ResNet50/N1C1/ResNet50_bs64_amp_fp16_SingleP_DP.sh @@ -1,6 +1,6 @@ model_item=ResNet50 bs_item=64 -fp_item=fp16 +fp_item=amp_fp16 run_process_type=SingleP run_mode=DP device_num=N1C1 diff --git a/test_tipc/static/ResNet50/N1C1/ResNet50_bs64_pure_fp16_SingleP_DP.sh b/test_tipc/static/ResNet50/N1C1/ResNet50_bs64_pure_fp16_SingleP_DP.sh new file mode 100644 index 0000000000000000000000000000000000000000..0c3ef054f4a85450f4566cdfa86066cab447ce57 --- /dev/null +++ b/test_tipc/static/ResNet50/N1C1/ResNet50_bs64_pure_fp16_SingleP_DP.sh @@ -0,0 +1,13 @@ +model_item=ResNet50 +bs_item=64 +fp_item=pure_fp16 +run_process_type=SingleP +run_mode=DP +device_num=N1C1 +max_epochs=1 +num_workers=8 + +# get data +bash test_tipc/static/${model_item}/benchmark_common/prepare.sh +# run +bash test_tipc/static/${model_item}/benchmark_common/run_benchmark.sh ${model_item} ${bs_item} ${fp_item} ${run_process_type} ${run_mode} ${device_num} ${max_epochs} ${num_workers} 2>&1; diff --git a/test_tipc/static/ResNet50/N1C8/ResNet50_bs128_amp_fp16_MultiP_DP.sh b/test_tipc/static/ResNet50/N1C8/ResNet50_bs128_amp_fp16_MultiP_DP.sh new file mode 100644 index 0000000000000000000000000000000000000000..5ee4b25a2b1ebca476c40e83dd492d73db74d145 --- /dev/null +++ b/test_tipc/static/ResNet50/N1C8/ResNet50_bs128_amp_fp16_MultiP_DP.sh @@ -0,0 +1,13 @@ +model_item=ResNet50 +bs_item=128 +fp_item=amp_fp16 +run_process_type=MultiP +run_mode=DP +device_num=N1C8 +max_epochs=8 +num_workers=8 + +# get data +bash test_tipc/static/${model_item}/benchmark_common/prepare.sh +# run +bash test_tipc/static/${model_item}/benchmark_common/run_benchmark.sh ${model_item} ${bs_item} ${fp_item} ${run_process_type} ${run_mode} ${device_num} ${max_epochs} ${num_workers} 2>&1; diff --git a/test_tipc/static/ResNet50/N1C8/ResNet50_bs128_fp32_MultiP_DP.sh b/test_tipc/static/ResNet50/N1C8/ResNet50_bs128_fp32_MultiP_DP.sh new file mode 100644 index 0000000000000000000000000000000000000000..eed2099c6f43ce1bac2fca31a9ccad3b1bc9c989 --- /dev/null +++ b/test_tipc/static/ResNet50/N1C8/ResNet50_bs128_fp32_MultiP_DP.sh @@ -0,0 +1,13 @@ +model_item=ResNet50 +bs_item=128 +fp_item=fp32 +run_process_type=MultiP +run_mode=DP +device_num=N1C8 +max_epochs=8 +num_workers=8 + +# get data +bash test_tipc/static/${model_item}/benchmark_common/prepare.sh +# run +bash test_tipc/static/${model_item}/benchmark_common/run_benchmark.sh ${model_item} ${bs_item} ${fp_item} ${run_process_type} ${run_mode} ${device_num} ${max_epochs} ${num_workers} 2>&1; diff --git a/test_tipc/static/ResNet50/N1C8/ResNet50_bs128_pure_fp16_MultiP_DP.sh b/test_tipc/static/ResNet50/N1C8/ResNet50_bs128_pure_fp16_MultiP_DP.sh new file mode 100644 index 0000000000000000000000000000000000000000..55efd4bd7302ef4ad4b29a2fe4ff0f1ddd3d8dc1 --- /dev/null +++ b/test_tipc/static/ResNet50/N1C8/ResNet50_bs128_pure_fp16_MultiP_DP.sh @@ -0,0 +1,13 @@ +model_item=ResNet50 +bs_item=128 +fp_item=pure_fp16 +run_process_type=MultiP +run_mode=DP +device_num=N1C8 +max_epochs=8 +num_workers=8 + +# get data +bash test_tipc/static/${model_item}/benchmark_common/prepare.sh +# run +bash test_tipc/static/${model_item}/benchmark_common/run_benchmark.sh ${model_item} ${bs_item} ${fp_item} ${run_process_type} ${run_mode} ${device_num} ${max_epochs} ${num_workers} 2>&1; diff --git a/test_tipc/static/ResNet50/N1C8/ResNet50_bs256_fp16_MultiP_DP.sh b/test_tipc/static/ResNet50/N1C8/ResNet50_bs256_amp_fp16_MultiP_DP.sh similarity index 95% rename from test_tipc/static/ResNet50/N1C8/ResNet50_bs256_fp16_MultiP_DP.sh rename to test_tipc/static/ResNet50/N1C8/ResNet50_bs256_amp_fp16_MultiP_DP.sh index d20792d947bb03f63d3329dad1b7179fc1ce8728..70d33e6ed1941f0903b36dda22bbbfed8d1b8164 100644 --- a/test_tipc/static/ResNet50/N1C8/ResNet50_bs256_fp16_MultiP_DP.sh +++ b/test_tipc/static/ResNet50/N1C8/ResNet50_bs256_amp_fp16_MultiP_DP.sh @@ -1,6 +1,6 @@ model_item=ResNet50 bs_item=256 -fp_item=fp16 +fp_item=amp_fp16 run_process_type=MultiP run_mode=DP device_num=N1C8 diff --git a/test_tipc/static/ResNet50/N1C8/ResNet50_bs256_pure_fp16_MultiP_DP.sh b/test_tipc/static/ResNet50/N1C8/ResNet50_bs256_pure_fp16_MultiP_DP.sh new file mode 100644 index 0000000000000000000000000000000000000000..ada8a0d2858d6b183f044711a2cfd14000c7ea90 --- /dev/null +++ b/test_tipc/static/ResNet50/N1C8/ResNet50_bs256_pure_fp16_MultiP_DP.sh @@ -0,0 +1,13 @@ +model_item=ResNet50 +bs_item=256 +fp_item=pure_fp16 +run_process_type=MultiP +run_mode=DP +device_num=N1C8 +max_epochs=8 +num_workers=8 + +# get data +bash test_tipc/static/${model_item}/benchmark_common/prepare.sh +# run +bash test_tipc/static/${model_item}/benchmark_common/run_benchmark.sh ${model_item} ${bs_item} ${fp_item} ${run_process_type} ${run_mode} ${device_num} ${max_epochs} ${num_workers} 2>&1; diff --git a/test_tipc/static/ResNet50/N1C8/ResNet50_bs64_fp16_MultiP_DP.sh b/test_tipc/static/ResNet50/N1C8/ResNet50_bs64_amp_fp16_MultiP_DP.sh similarity index 95% rename from test_tipc/static/ResNet50/N1C8/ResNet50_bs64_fp16_MultiP_DP.sh rename to test_tipc/static/ResNet50/N1C8/ResNet50_bs64_amp_fp16_MultiP_DP.sh index 7edf4e4210bcb338e61848952dc3cbde4c4f4ebb..8141d002853eaa5d7013615e5d80703196c9d866 100644 --- a/test_tipc/static/ResNet50/N1C8/ResNet50_bs64_fp16_MultiP_DP.sh +++ b/test_tipc/static/ResNet50/N1C8/ResNet50_bs64_amp_fp16_MultiP_DP.sh @@ -1,6 +1,6 @@ model_item=ResNet50 bs_item=64 -fp_item=fp16 +fp_item=amp_fp16 run_process_type=MultiP run_mode=DP device_num=N1C8 diff --git a/test_tipc/static/ResNet50/N1C8/ResNet50_bs64_pure_fp16_MultiP_DP.sh b/test_tipc/static/ResNet50/N1C8/ResNet50_bs64_pure_fp16_MultiP_DP.sh new file mode 100644 index 0000000000000000000000000000000000000000..acf8dcdbe84171588b3f54d8a11424f0e6e342fd --- /dev/null +++ b/test_tipc/static/ResNet50/N1C8/ResNet50_bs64_pure_fp16_MultiP_DP.sh @@ -0,0 +1,13 @@ +model_item=ResNet50 +bs_item=64 +fp_item=pure_fp16 +run_process_type=MultiP +run_mode=DP +device_num=N1C8 +max_epochs=8 +num_workers=8 + +# get data +bash test_tipc/static/${model_item}/benchmark_common/prepare.sh +# run +bash test_tipc/static/${model_item}/benchmark_common/run_benchmark.sh ${model_item} ${bs_item} ${fp_item} ${run_process_type} ${run_mode} ${device_num} ${max_epochs} ${num_workers} 2>&1; diff --git a/test_tipc/static/ResNet50/N4C32/ResNet50_bs128_amp_fp16_MultiP_DP.sh b/test_tipc/static/ResNet50/N4C32/ResNet50_bs128_amp_fp16_MultiP_DP.sh new file mode 100644 index 0000000000000000000000000000000000000000..195541240b7d055774834e2d53ad7d8dc0bbdbf2 --- /dev/null +++ b/test_tipc/static/ResNet50/N4C32/ResNet50_bs128_amp_fp16_MultiP_DP.sh @@ -0,0 +1,13 @@ +model_item=ResNet50 +bs_item=128 +fp_item=amp_fp16 +run_process_type=MultiP +run_mode=DP +device_num=N4C32 +max_epochs=32 +num_workers=8 + +# get data +bash test_tipc/static/${model_item}/benchmark_common/prepare.sh +# run +bash test_tipc/static/${model_item}/benchmark_common/run_benchmark.sh ${model_item} ${bs_item} ${fp_item} ${run_process_type} ${run_mode} ${device_num} ${max_epochs} ${num_workers} 2>&1; diff --git a/test_tipc/static/ResNet50/N4C32/ResNet50_bs128_fp32_MultiP_DP.sh b/test_tipc/static/ResNet50/N4C32/ResNet50_bs128_fp32_MultiP_DP.sh new file mode 100644 index 0000000000000000000000000000000000000000..3b55490639acaf46d72a6d2ec352645017e18c77 --- /dev/null +++ b/test_tipc/static/ResNet50/N4C32/ResNet50_bs128_fp32_MultiP_DP.sh @@ -0,0 +1,13 @@ +model_item=ResNet50 +bs_item=128 +fp_item=fp32 +run_process_type=MultiP +run_mode=DP +device_num=N4C32 +max_epochs=32 +num_workers=8 + +# get data +bash test_tipc/static/${model_item}/benchmark_common/prepare.sh +# run +bash test_tipc/static/${model_item}/benchmark_common/run_benchmark.sh ${model_item} ${bs_item} ${fp_item} ${run_process_type} ${run_mode} ${device_num} ${max_epochs} ${num_workers} 2>&1; diff --git a/test_tipc/static/ResNet50/N4C32/ResNet50_bs128_pure_fp16_MultiP_DP.sh b/test_tipc/static/ResNet50/N4C32/ResNet50_bs128_pure_fp16_MultiP_DP.sh new file mode 100644 index 0000000000000000000000000000000000000000..95ed49acf2ed32841b3a76123f19620ac9a930de --- /dev/null +++ b/test_tipc/static/ResNet50/N4C32/ResNet50_bs128_pure_fp16_MultiP_DP.sh @@ -0,0 +1,13 @@ +model_item=ResNet50 +bs_item=128 +fp_item=pure_fp16 +run_process_type=MultiP +run_mode=DP +device_num=N4C32 +max_epochs=32 +num_workers=8 + +# get data +bash test_tipc/static/${model_item}/benchmark_common/prepare.sh +# run +bash test_tipc/static/${model_item}/benchmark_common/run_benchmark.sh ${model_item} ${bs_item} ${fp_item} ${run_process_type} ${run_mode} ${device_num} ${max_epochs} ${num_workers} 2>&1; diff --git a/test_tipc/static/ResNet50/N4C32/ResNet50_bs256_amp_fp16_MultiP_DP.sh b/test_tipc/static/ResNet50/N4C32/ResNet50_bs256_amp_fp16_MultiP_DP.sh new file mode 100644 index 0000000000000000000000000000000000000000..4997154dbf306db7f9ed7b077926e32836ed35c6 --- /dev/null +++ b/test_tipc/static/ResNet50/N4C32/ResNet50_bs256_amp_fp16_MultiP_DP.sh @@ -0,0 +1,13 @@ +model_item=ResNet50 +bs_item=256 +fp_item=amp_fp16 +run_process_type=MultiP +run_mode=DP +device_num=N4C32 +max_epochs=32 +num_workers=8 + +# get data +bash test_tipc/static/${model_item}/benchmark_common/prepare.sh +# run +bash test_tipc/static/${model_item}/benchmark_common/run_benchmark.sh ${model_item} ${bs_item} ${fp_item} ${run_process_type} ${run_mode} ${device_num} ${max_epochs} ${num_workers} 2>&1; diff --git a/test_tipc/static/ResNet50/N4C32/ResNet50_bs256_fp32_MultiP_DP.sh b/test_tipc/static/ResNet50/N4C32/ResNet50_bs256_fp32_MultiP_DP.sh new file mode 100644 index 0000000000000000000000000000000000000000..7ff8bf640c8fcbab35b67b36a751da11bb84195b --- /dev/null +++ b/test_tipc/static/ResNet50/N4C32/ResNet50_bs256_fp32_MultiP_DP.sh @@ -0,0 +1,13 @@ +model_item=ResNet50 +bs_item=256 +fp_item=fp32 +run_process_type=MultiP +run_mode=DP +device_num=N4C32 +max_epochs=32 +num_workers=8 + +# get data +bash test_tipc/static/${model_item}/benchmark_common/prepare.sh +# run +bash test_tipc/static/${model_item}/benchmark_common/run_benchmark.sh ${model_item} ${bs_item} ${fp_item} ${run_process_type} ${run_mode} ${device_num} ${max_epochs} ${num_workers} 2>&1; diff --git a/test_tipc/static/ResNet50/N4C32/ResNet50_bs256_pure_fp16_MultiP_DP.sh b/test_tipc/static/ResNet50/N4C32/ResNet50_bs256_pure_fp16_MultiP_DP.sh new file mode 100644 index 0000000000000000000000000000000000000000..a43cdf179088db025043228741da4a079e0a47de --- /dev/null +++ b/test_tipc/static/ResNet50/N4C32/ResNet50_bs256_pure_fp16_MultiP_DP.sh @@ -0,0 +1,13 @@ +model_item=ResNet50 +bs_item=256 +fp_item=pure_fp16 +run_process_type=MultiP +run_mode=DP +device_num=N4C32 +max_epochs=32 +num_workers=8 + +# get data +bash test_tipc/static/${model_item}/benchmark_common/prepare.sh +# run +bash test_tipc/static/${model_item}/benchmark_common/run_benchmark.sh ${model_item} ${bs_item} ${fp_item} ${run_process_type} ${run_mode} ${device_num} ${max_epochs} ${num_workers} 2>&1; diff --git a/test_tipc/static/ResNet50/N4C32/ResNet50_bs64_amp_fp16_MultiP_DP.sh b/test_tipc/static/ResNet50/N4C32/ResNet50_bs64_amp_fp16_MultiP_DP.sh new file mode 100644 index 0000000000000000000000000000000000000000..6d9b26a9cad46c8ec346b64d30bcba8b941e40d2 --- /dev/null +++ b/test_tipc/static/ResNet50/N4C32/ResNet50_bs64_amp_fp16_MultiP_DP.sh @@ -0,0 +1,13 @@ +model_item=ResNet50 +bs_item=64 +fp_item=amp_fp16 +run_process_type=MultiP +run_mode=DP +device_num=N4C32 +max_epochs=32 +num_workers=8 + +# get data +bash test_tipc/static/${model_item}/benchmark_common/prepare.sh +# run +bash test_tipc/static/${model_item}/benchmark_common/run_benchmark.sh ${model_item} ${bs_item} ${fp_item} ${run_process_type} ${run_mode} ${device_num} ${max_epochs} ${num_workers} 2>&1; diff --git a/test_tipc/static/ResNet50/N4C32/ResNet50_bs64_fp32_MultiP_DP.sh b/test_tipc/static/ResNet50/N4C32/ResNet50_bs64_fp32_MultiP_DP.sh new file mode 100644 index 0000000000000000000000000000000000000000..54b19abcdf841c282f4ab1552ca1ea0476b6799b --- /dev/null +++ b/test_tipc/static/ResNet50/N4C32/ResNet50_bs64_fp32_MultiP_DP.sh @@ -0,0 +1,13 @@ +model_item=ResNet50 +bs_item=64 +fp_item=fp32 +run_process_type=MultiP +run_mode=DP +device_num=N4C32 +max_epochs=32 +num_workers=8 + +# get data +bash test_tipc/static/${model_item}/benchmark_common/prepare.sh +# run +bash test_tipc/static/${model_item}/benchmark_common/run_benchmark.sh ${model_item} ${bs_item} ${fp_item} ${run_process_type} ${run_mode} ${device_num} ${max_epochs} ${num_workers} 2>&1; diff --git a/test_tipc/static/ResNet50/N4C32/ResNet50_bs64_pure_fp16_MultiP_DP.sh b/test_tipc/static/ResNet50/N4C32/ResNet50_bs64_pure_fp16_MultiP_DP.sh new file mode 100644 index 0000000000000000000000000000000000000000..e3c3daaf5716862e3d747342f65d4b3820a3f20f --- /dev/null +++ b/test_tipc/static/ResNet50/N4C32/ResNet50_bs64_pure_fp16_MultiP_DP.sh @@ -0,0 +1,13 @@ +model_item=ResNet50 +bs_item=64 +fp_item=pure_fp16 +run_process_type=MultiP +run_mode=DP +device_num=N4C32 +max_epochs=32 +num_workers=8 + +# get data +bash test_tipc/static/${model_item}/benchmark_common/prepare.sh +# run +bash test_tipc/static/${model_item}/benchmark_common/run_benchmark.sh ${model_item} ${bs_item} ${fp_item} ${run_process_type} ${run_mode} ${device_num} ${max_epochs} ${num_workers} 2>&1; diff --git a/test_tipc/static/ResNet50/benchmark_common/run_benchmark.sh b/test_tipc/static/ResNet50/benchmark_common/run_benchmark.sh index 2c3e4e1e30c7770b75e0e6401c973ded04de227e..b66602ea61b5e6cef7564c834fa03fd7692f404d 100644 --- a/test_tipc/static/ResNet50/benchmark_common/run_benchmark.sh +++ b/test_tipc/static/ResNet50/benchmark_common/run_benchmark.sh @@ -35,8 +35,10 @@ function _train(){ if [ ${fp_item} = "fp32" ]; then config_file="-c ppcls/configs/ImageNet/ResNet/ResNet50.yaml" - else + elif [ ${fp_item} = "amp_fp16" ]; then config_file="-c ppcls/configs/ImageNet/ResNet/ResNet50_amp_O1.yaml" + elif [ ${fp_item} = "pure_fp16" ]; then + config_file="-c ppcls/configs/ImageNet/ResNet/ResNet50_amp_O2.yaml" fi if [ ${profiling} = "false" ]; then profiling_config=""