diff --git a/TensorFlow2/built-in/nlp/bert-squad_ID1566_for_TensorFlow2.X/test/set_conda.sh b/TensorFlow2/built-in/nlp/bert-squad_ID1566_for_TensorFlow2.X/test/set_conda.sh new file mode 100644 index 0000000000000000000000000000000000000000..55087d8622f46e055bb105a2acd1fe7006d4af07 --- /dev/null +++ b/TensorFlow2/built-in/nlp/bert-squad_ID1566_for_TensorFlow2.X/test/set_conda.sh @@ -0,0 +1,2 @@ +export PATH=/home/anaconda3/bin:$PATH +export LD_LIBRARY_PATH=/home/anaconda3/lib:$LD_LIBRARY_PATH \ No newline at end of file diff --git a/TensorFlow2/built-in/nlp/bert-squad_ID1566_for_TensorFlow2.X/test/train_performance_1p.sh b/TensorFlow2/built-in/nlp/bert-squad_ID1566_for_TensorFlow2.X/test/train_performance_1p.sh index 29b4d082a12817c6d9163fd448c47722f94d3a42..c457b7d200ab5bcf39af014ea1e64a1ff7e41f7b 100644 --- a/TensorFlow2/built-in/nlp/bert-squad_ID1566_for_TensorFlow2.X/test/train_performance_1p.sh +++ b/TensorFlow2/built-in/nlp/bert-squad_ID1566_for_TensorFlow2.X/test/train_performance_1p.sh @@ -82,6 +82,10 @@ do fusion_off_file=`echo ${para#*=}` elif [[ $para == --auto_tune* ]];then auto_tune=`echo ${para#*=}` + elif [[ $para == --conda_name* ]];then + conda_name=`echo ${para#*=}` + source $cur_path/test/set_conda.sh + source activate $conda_name fi done ############维测参数##############