wangmin
5 years ago
1 changed files with 19 additions and 21 deletions
Unified View
Diff Options
+ 19
- 21
model_zoo/official/cv/resnet_thor/README.md
View File
| @@ -79,27 +79,25 @@ sh run_eval_gpu.sh [DATASET_PATH] [CHECKPOINT_PATH] | |||||
| ### Script Code Structure | ### Script Code Structure | ||||
| ```shell | ```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 | ### Script Parameters | ||||