fix fp16 benchmark train

This commit is contained in:
LDOUBLEV
2022-07-13 18:07:06 +08:00
parent 2e54aa6809
commit 4c24368b7d
2 changed files with 3 additions and 3 deletions
+2 -2
View File
@@ -139,8 +139,8 @@ else
device_num=${params_list[4]}
IFS=";"
if [ ${precision} = "null" ];then
precision="fp32"
if [ ${precision} = "fp16" ];then
precision="amp"
fi
fp_items_list=($precision)
+1 -1
View File
@@ -265,7 +265,7 @@ else
if [ ${run_train} = "null" ]; then
continue
fi
set_autocast=$(func_set_params "${autocast_key}" "${autocast}")
set_epoch=$(func_set_params "${epoch_key}" "${epoch_num}")
set_pretrain=$(func_set_params "${pretrain_model_key}" "${pretrain_model_value}")
set_batchsize=$(func_set_params "${train_batch_key}" "${train_batch_value}")