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

fix fp16 benchmark train

上级 2e54aa68
......@@ -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)
......
......@@ -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}")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册