diff --git a/test_tipc/benchmark_train.sh b/test_tipc/benchmark_train.sh index 19c288ff95b90c4ca712faf2717b8611ad7a44a6..ca85b909051e19c6405de356b326bfc462369612 100644 --- a/test_tipc/benchmark_train.sh +++ b/test_tipc/benchmark_train.sh @@ -156,6 +156,7 @@ else if [ ${precision} = "fp16" ];then precision="amp" fi + epoch=$(set_dynamic_epoch $device_num $epoch) fp_items_list=($precision) batch_size_list=($batch_size)