|
|
@ -20,10 +20,10 @@ train_use_gpu_value=$(func_parser_value "${lines[4]}")
|
|
|
|
autocast_list=$(func_parser_value "${lines[5]}")
|
|
|
|
autocast_list=$(func_parser_value "${lines[5]}")
|
|
|
|
autocast_key=$(func_parser_key "${lines[5]}")
|
|
|
|
autocast_key=$(func_parser_key "${lines[5]}")
|
|
|
|
epoch_key=$(func_parser_key "${lines[6]}")
|
|
|
|
epoch_key=$(func_parser_key "${lines[6]}")
|
|
|
|
epoch_num=$(func_parser_params "${lines[6]}" "${MODE}")
|
|
|
|
epoch_num=`echo "${lines[6]}" | tr -cd "[0-9]" `
|
|
|
|
save_model_key=$(func_parser_key "${lines[7]}")
|
|
|
|
save_model_key=$(func_parser_key "${lines[7]}")
|
|
|
|
train_batch_key=$(func_parser_key "${lines[8]}")
|
|
|
|
train_batch_key=$(func_parser_key "${lines[8]}")
|
|
|
|
train_batch_value=$(func_parser_params "${lines[8]}" "${MODE}")
|
|
|
|
train_batch_value=`echo "${lines[8]}" | tr -cd "[0-9]" `
|
|
|
|
pretrain_model_key=$(func_parser_key "${lines[9]}")
|
|
|
|
pretrain_model_key=$(func_parser_key "${lines[9]}")
|
|
|
|
pretrain_model_value=$(func_parser_value "${lines[9]}")
|
|
|
|
pretrain_model_value=$(func_parser_value "${lines[9]}")
|
|
|
|
train_model_name=$(func_parser_value "${lines[10]}")
|
|
|
|
train_model_name=$(func_parser_value "${lines[10]}")
|
|
|
@ -374,4 +374,4 @@ else
|
|
|
|
done # done with: for trainer in ${trainer_list[*]}; do
|
|
|
|
done # done with: for trainer in ${trainer_list[*]}; do
|
|
|
|
done # done with: for autocast in ${autocast_list[*]}; do
|
|
|
|
done # done with: for autocast in ${autocast_list[*]}; do
|
|
|
|
done # done with: for gpu in ${gpu_list[*]}; do
|
|
|
|
done # done with: for gpu in ${gpu_list[*]}; do
|
|
|
|
fi # end if [ ${MODE} = "infer" ]; then
|
|
|
|
fi # end if [ ${MODE} = "infer" ]; then
|
|
|
|