xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!5100 fix thor optimizer readme file for resnet50 

Merge pull request !5100 from wangmin0104/master
This commit is contained in:
mindspore-ci-bot 2020-08-25 17:23:05 +08:00 committed by Gitee
parent b3f0899da2 2a2c944d19
commit ac239b6506
1 changed files with 19 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
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
```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