fix log name
@@ -150,10 +150,16 @@ fi
IFS="|"
for batch_size in ${batch_size_list[*]}; do
- for precision in ${fp_items_list[*]}; do
+ for train_precision in ${fp_items_list[*]}; do
for device_num in ${device_num_list[*]}; do
# sed batchsize and precision
- func_sed_params "$FILENAME" "${line_precision}" "$precision"
+ if [ ${train_precision} = "amp" ];then
+ precision="fp16"
+ else
+ precision="fp32"
+ fi
+
+ func_sed_params "$FILENAME" "${line_precision}" "$train_precision"
func_sed_params "$FILENAME" "${line_batchsize}" "$MODE=$batch_size"
func_sed_params "$FILENAME" "${line_epoch}" "$MODE=$epoch"
gpu_id=$(set_gpu_id $device_num)
