diff --git a/TensorFlow/built-in/cv/image_classification/ResNet50_ID0058_for_TensorFlow/README.md b/TensorFlow/built-in/cv/image_classification/ResNet50_ID0058_for_TensorFlow/README.md index 5ff826c98726177dc54ccbb731785522c430e349..07b2923d9d3afaae4c387edc42b6349a7beea84a 100644 --- a/TensorFlow/built-in/cv/image_classification/ResNet50_ID0058_for_TensorFlow/README.md +++ b/TensorFlow/built-in/cv/image_classification/ResNet50_ID0058_for_TensorFlow/README.md @@ -139,16 +139,28 @@ python3.7 ../src/mains/res50.py \ ``` More parameters are in --config_file under src/configs. ```shell -cd /path/to/Modelzoo_Resnet50_HC/scripts +cd /path/to/ResNet50_ID0058_for_TensorFlow/scripts bash train_1p.sh ``` 8P training is similar to the former: ```shell -cd /path/to/Modelzoo_Resnet50_HC/scripts +cd /path/to/ResNet50_ID0058_for_TensorFlow/scripts bash train_8p.sh ``` +We recommend using the updated training shell instructions in `/test`,examples: +``` +cd /path/to/ResNet50_ID0058_for_TensorFlow/test +bash train_performance_bs32_1p.sh +``` +8P training is also similar to the former: +``` +cd /path/to/ResNet50_ID0058_for_TensorFlow/test +bash train_performance_bs32_8p.sh +``` + + ##### ascend_distribute Training training shell `./test/train_performance_distribute.sh`, change from `./test/train_performance_bs256_hw192_8p.sh` diff --git a/TensorFlow/built-in/cv/image_classification/ResNet50_ID0058_for_TensorFlow/scripts/train_1p.sh b/TensorFlow/built-in/cv/image_classification/ResNet50_ID0058_for_TensorFlow/scripts/train_1p.sh index 3f5c4013a0655dd24f5eebd2412c4411cf63af13..e4c31f44fc64906aec8e0a4f257157b5303296b4 100644 --- a/TensorFlow/built-in/cv/image_classification/ResNet50_ID0058_for_TensorFlow/scripts/train_1p.sh +++ b/TensorFlow/built-in/cv/image_classification/ResNet50_ID0058_for_TensorFlow/scripts/train_1p.sh @@ -19,7 +19,8 @@ mkdir -p ${currentDir}/d_solution/ckpt${DEVICE_ID} env > ${currentDir}/log/env_${device_id}.log #start exec -python3.7 ../src/mains/res50.py \ +cd ${currentDir}/../ +python3.7 ${currentDir}/../src/mains/res50.py \ --config_file=res50_256bs_1p \ --max_train_steps=1000 \ --iterations_per_loop=100 \ diff --git a/TensorFlow/built-in/cv/image_classification/ResNet50_ID0058_for_TensorFlow/scripts/train_sample.sh b/TensorFlow/built-in/cv/image_classification/ResNet50_ID0058_for_TensorFlow/scripts/train_sample.sh index 9b655c0fbb0d48a7ae7a5c31165c5e4dd753a7a7..3258fc5e1e602d799b93268cb4a5f509f8735928 100644 --- a/TensorFlow/built-in/cv/image_classification/ResNet50_ID0058_for_TensorFlow/scripts/train_sample.sh +++ b/TensorFlow/built-in/cv/image_classification/ResNet50_ID0058_for_TensorFlow/scripts/train_sample.sh @@ -21,7 +21,8 @@ mkdir -p ${currentDir}/d_solution/ckpt${DEVICE_ID} env > ${currentDir}/log/env_${device_id}.log #start exec -python3.7 ../src/mains/res50.py \ +cd ${currentDir}/../ +python3.7 ${currentDir}/../src/mains/res50.py \ --config_file=res50_256bs_8p \ --max_train_steps=1000 \ --iterations_per_loop=100 \