diff --git a/TensorFlow2/built-in/cv/image_classification/ResNet50_ID0360_for_TensorFlow2.X/test/train_full_8p_256bs_SGD.sh b/TensorFlow2/built-in/cv/image_classification/ResNet50_ID0360_for_TensorFlow2.X/test/train_full_8p_256bs_SGD.sh index fe47982c0145e12e5617f52c9a518d42eb91ab06..90addd954541f221ee5ffdd7b190339ecd94bb19 100644 --- a/TensorFlow2/built-in/cv/image_classification/ResNet50_ID0360_for_TensorFlow2.X/test/train_full_8p_256bs_SGD.sh +++ b/TensorFlow2/built-in/cv/image_classification/ResNet50_ID0360_for_TensorFlow2.X/test/train_full_8p_256bs_SGD.sh @@ -77,6 +77,10 @@ do elif [[ $para == --bind_core* ]]; then bind_core=`echo ${para#*=}` name_bind="_bindcore" + elif [[ $para == --conda_name* ]];then + conda_name=`echo ${para#*=}` + source set_conda.sh + source activate $conda_name fi done diff --git a/TensorFlow2/built-in/nlp/bert-squad_ID1566_for_TensorFlow2.X/test/train_full_squad1.1_base_8p.sh b/TensorFlow2/built-in/nlp/bert-squad_ID1566_for_TensorFlow2.X/test/train_full_squad1.1_base_8p.sh index 84cb15e8662767b431f76287a47b5d236806e448..989b005a04564fd80d71e40e329f2ce23ebaa390 100644 --- a/TensorFlow2/built-in/nlp/bert-squad_ID1566_for_TensorFlow2.X/test/train_full_squad1.1_base_8p.sh +++ b/TensorFlow2/built-in/nlp/bert-squad_ID1566_for_TensorFlow2.X/test/train_full_squad1.1_base_8p.sh @@ -83,6 +83,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 ############维测参数##############