Skip to content

Commit 23d56b7

Browse files
author
Swetha Mandava
committed
replacing / with - in init_checkpoint string for printing
1 parent b4384a5 commit 23d56b7

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

TensorFlow/LanguageModeling/BERT/scripts/run_glue_inference.sh

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -45,9 +45,10 @@ else
4545
use_xla_tag=""
4646
fi
4747

48-
48+
num_gpu=1
49+
ckpt_str=${init_checkpoint//\//-}
4950
export GBS=$(expr $batch_size \* $num_gpu)
50-
printf -v TAG "tf_bert_finetuning_glue_%s_inf_%s_%s_gbs%d_ckpt_%s" "$task_name" "$bert_model" "$precision" $GBS "$init_checkpoint"
51+
printf -v TAG "tf_bert_finetuning_glue_%s_inf_%s_%s_gbs%d_ckpt_%s" "$task_name" "$bert_model" "$precision" $GBS "$ckpt_str"
5152
DATESTAMP=`date +'%y%m%d%H%M%S'`
5253
#Edit to save logs & checkpoints in a different directory
5354
RESULTS_DIR=/results
@@ -75,4 +76,4 @@ $mpi_command python run_classifier.py \
7576
--doc_stride=$doc_stride \
7677
--output_dir=$RESULTS_DIR \
7778
--horovod "$use_fp16" \
78-
$use_xla_tag |& tee $LOGFILE
79+
$use_xla_tag |& tee $LOGFILE

TensorFlow/LanguageModeling/BERT/scripts/run_squad_inference.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,8 @@ else
5454
use_xla_tag=""
5555
fi
5656

57-
printf -v TAG "tf_bert_finetuning_squad_%s_inf_%s_gbs%d_ckpt_%s" "$bert_model" "$precision" $batch_size "$init_checkpoint"
57+
ckpt_str=${init_checkpoint//\//-}
58+
printf -v TAG "tf_bert_finetuning_squad_%s_inf_%s_gbs%d_ckpt_%s" "$bert_model" "$precision" $batch_size "$ckpt_str"
5859
DATESTAMP=`date +'%y%m%d%H%M%S'`
5960
#Edit to save logs & checkpoints in a different directory
6061
RESULTS_DIR=/results

0 commit comments

Comments
 (0)