提交 4c24368b 编写于 作者: L LDOUBLEV

fix fp16 benchmark train

上级 2e54aa68
...@@ -139,8 +139,8 @@ else ...@@ -139,8 +139,8 @@ else
device_num=${params_list[4]} device_num=${params_list[4]}
IFS=";" IFS=";"
if [ ${precision} = "null" ];then if [ ${precision} = "fp16" ];then
precision="fp32" precision="amp"
fi fi
fp_items_list=($precision) fp_items_list=($precision)
......
...@@ -265,7 +265,7 @@ else ...@@ -265,7 +265,7 @@ else
if [ ${run_train} = "null" ]; then if [ ${run_train} = "null" ]; then
continue continue
fi fi
set_autocast=$(func_set_params "${autocast_key}" "${autocast}")
set_epoch=$(func_set_params "${epoch_key}" "${epoch_num}") set_epoch=$(func_set_params "${epoch_key}" "${epoch_num}")
set_pretrain=$(func_set_params "${pretrain_model_key}" "${pretrain_model_value}") set_pretrain=$(func_set_params "${pretrain_model_key}" "${pretrain_model_value}")
set_batchsize=$(func_set_params "${train_batch_key}" "${train_batch_value}") set_batchsize=$(func_set_params "${train_batch_key}" "${train_batch_value}")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册