diff --git a/TensorFlow/built-in/cv/detection/FasterRcnn_resnet50_ID0010_for_TensorFlow/FasterRcnn/configs/mask_rcnn_config.py b/TensorFlow/built-in/cv/detection/FasterRcnn_resnet50_ID0010_for_TensorFlow/FasterRcnn/configs/mask_rcnn_config.py index 862da6beeb628e213f552384b4763e9f40d56da3..fdbd893ee1974dda34f0de398876bf74cff4fbb2 100644 --- a/TensorFlow/built-in/cv/detection/FasterRcnn_resnet50_ID0010_for_TensorFlow/FasterRcnn/configs/mask_rcnn_config.py +++ b/TensorFlow/built-in/cv/detection/FasterRcnn_resnet50_ID0010_for_TensorFlow/FasterRcnn/configs/mask_rcnn_config.py @@ -34,7 +34,7 @@ MASK_RCNN_CFG = { 'num_gpus': 1, # runtime parameters 'transpose_input': True, - 'iterations_per_loop': 1000, + 'iterations_per_loop': 1, 'num_cores': 8, 'use_tpu': False, # input preprocessing parameters diff --git a/TensorFlow2/built-in/cv/image_classification/ResNet50_ID0360_for_TensorFlow2.X/test/set_conda.sh b/TensorFlow2/built-in/cv/image_classification/ResNet50_ID0360_for_TensorFlow2.X/test/set_conda.sh new file mode 100644 index 0000000000000000000000000000000000000000..55087d8622f46e055bb105a2acd1fe7006d4af07 --- /dev/null +++ b/TensorFlow2/built-in/cv/image_classification/ResNet50_ID0360_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/cv/image_classification/ResNet50_ID0360_for_TensorFlow2.X/test/train_performance_8p_256bs_SGD.sh b/TensorFlow2/built-in/cv/image_classification/ResNet50_ID0360_for_TensorFlow2.X/test/train_performance_8p_256bs_SGD.sh index 768ba1b7ab663a4608a933f3f1ae72cb996a7551..7ef5bbe3402dafaa12969d7d483cf9ecc18da926 100644 --- a/TensorFlow2/built-in/cv/image_classification/ResNet50_ID0360_for_TensorFlow2.X/test/train_performance_8p_256bs_SGD.sh +++ b/TensorFlow2/built-in/cv/image_classification/ResNet50_ID0360_for_TensorFlow2.X/test/train_performance_8p_256bs_SGD.sh @@ -76,6 +76,11 @@ 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 fi done