From 23d56b7ceeaa80828262bf886bdc54915925f058 Mon Sep 17 00:00:00 2001 From: Swetha Mandava Date: Fri, 13 Dec 2019 19:00:40 -0800 Subject: [PATCH] replacing / with - in init_checkpoint string for printing --- .../LanguageModeling/BERT/scripts/run_glue_inference.sh | 7 ++++--- .../LanguageModeling/BERT/scripts/run_squad_inference.sh | 3 ++- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/TensorFlow/LanguageModeling/BERT/scripts/run_glue_inference.sh b/TensorFlow/LanguageModeling/BERT/scripts/run_glue_inference.sh index 6e7e0e75..29fe4823 100644 --- a/TensorFlow/LanguageModeling/BERT/scripts/run_glue_inference.sh +++ b/TensorFlow/LanguageModeling/BERT/scripts/run_glue_inference.sh @@ -45,9 +45,10 @@ else use_xla_tag="" fi - +num_gpu=1 +ckpt_str=${init_checkpoint//\//-} export GBS=$(expr $batch_size \* $num_gpu) -printf -v TAG "tf_bert_finetuning_glue_%s_inf_%s_%s_gbs%d_ckpt_%s" "$task_name" "$bert_model" "$precision" $GBS "$init_checkpoint" +printf -v TAG "tf_bert_finetuning_glue_%s_inf_%s_%s_gbs%d_ckpt_%s" "$task_name" "$bert_model" "$precision" $GBS "$ckpt_str" DATESTAMP=`date +'%y%m%d%H%M%S'` #Edit to save logs & checkpoints in a different directory RESULTS_DIR=/results @@ -75,4 +76,4 @@ $mpi_command python run_classifier.py \ --doc_stride=$doc_stride \ --output_dir=$RESULTS_DIR \ --horovod "$use_fp16" \ - $use_xla_tag |& tee $LOGFILE \ No newline at end of file + $use_xla_tag |& tee $LOGFILE diff --git a/TensorFlow/LanguageModeling/BERT/scripts/run_squad_inference.sh b/TensorFlow/LanguageModeling/BERT/scripts/run_squad_inference.sh index 2a8aa2cc..a6b64361 100755 --- a/TensorFlow/LanguageModeling/BERT/scripts/run_squad_inference.sh +++ b/TensorFlow/LanguageModeling/BERT/scripts/run_squad_inference.sh @@ -54,7 +54,8 @@ else use_xla_tag="" fi -printf -v TAG "tf_bert_finetuning_squad_%s_inf_%s_gbs%d_ckpt_%s" "$bert_model" "$precision" $batch_size "$init_checkpoint" +ckpt_str=${init_checkpoint//\//-} +printf -v TAG "tf_bert_finetuning_squad_%s_inf_%s_gbs%d_ckpt_%s" "$bert_model" "$precision" $batch_size "$ckpt_str" DATESTAMP=`date +'%y%m%d%H%M%S'` #Edit to save logs & checkpoints in a different directory RESULTS_DIR=/results