File tree Expand file tree Collapse file tree 2 files changed +6
-4
lines changed
TensorFlow/LanguageModeling/BERT/scripts Expand file tree Collapse file tree 2 files changed +6
-4
lines changed Original file line number Diff line number Diff line change 4545 use_xla_tag=" "
4646fi
4747
48-
48+ num_gpu=1
49+ ckpt_str=${init_checkpoint// \/ / -}
4950export 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 "
5152DATESTAMP=` date +' %y%m%d%H%M%S' `
5253# Edit to save logs & checkpoints in a different directory
5354RESULTS_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
Original file line number Diff line number Diff line change 5454 use_xla_tag=" "
5555fi
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 "
5859DATESTAMP=` date +' %y%m%d%H%M%S' `
5960# Edit to save logs & checkpoints in a different directory
6061RESULTS_DIR=/results
You can’t perform that action at this time.
0 commit comments