From 2a2c944d1998e5764d4e7bbe3e97c4300528c5c7 Mon Sep 17 00:00:00 2001 From: wangmin Date: Tue, 25 Aug 2020 09:16:25 +0800 Subject: [PATCH] fix thor optimizer readme for resnet50 --- model_zoo/official/cv/resnet_thor/README.md | 40 ++++++++++----------- 1 file changed, 19 insertions(+), 21 deletions(-) diff --git a/model_zoo/official/cv/resnet_thor/README.md b/model_zoo/official/cv/resnet_thor/README.md index f6de81316..d8f8d8a42 100644 --- a/model_zoo/official/cv/resnet_thor/README.md +++ b/model_zoo/official/cv/resnet_thor/README.md @@ -79,27 +79,25 @@ sh run_eval_gpu.sh [DATASET_PATH] [CHECKPOINT_PATH] ### Script Code Structure ```shell -├── model_zoo - ├──README.md # descriptions about all the models - ├── resnet_thor - ├── README.md # descriptions about resnet_thor - ├── scripts - │ ├── run_distribute_train.sh # launch distributed training for Ascend - │ └── run_eval.sh # launch infering for Ascend - │ ├── run_distribute_train_gpu.sh # launch distributed training for GPU - │ └── run_eval_gpu.sh # launch infering for GPU - ├──src - │ ├── crossentropy.py # CrossEntropy loss function - │ ├── config.py # parameter configuration - │ ├── dataset_helper.py # dataset help for minddata dataset - │ ├── grad_reducer_thor.py # grad reducer for thor - │ ├── model_thor.py # model for train - │ ├── resnet_thor.py # resnet50_thor backone - │ ├── thor.py # thor optimizer - │ ├── thor_layer.py # thor layer - │ └── dataset.py # data preprocessing - ├── eval.py # infer script - └── train.py # train script +└── resnet_thor + ├── README.md # descriptions about resnet_thor + ├── scripts + │ ├── run_distribute_train.sh # launch distributed training for Ascend + │ └── run_eval.sh # launch inference for Ascend + │ ├── run_distribute_train_gpu.sh # launch distributed training for GPU + │ └── run_eval_gpu.sh # launch inference for GPU + ├──src + │ ├── crossentropy.py # CrossEntropy loss function + │ ├── config.py # parameter configuration + │ ├── dataset_helper.py # dataset help for minddata dataset + │ ├── grad_reducer_thor.py # grad reducer for thor + │ ├── model_thor.py # model for train + │ ├── resnet_thor.py # resnet50_thor backone + │ ├── thor.py # thor optimizer + │ ├── thor_layer.py # thor layer + │ └── dataset.py # data preprocessing + ├── eval.py # infer script + └── train.py # train script ``` ### Script Parameters -- GitLab