Browse Source

fix thor optimizer readme for resnet50

tags/v1.0.0
wangmin 5 years ago
parent
commit
2a2c944d19
1 changed files with 19 additions and 21 deletions
  1. +19
    -21
      model_zoo/official/cv/resnet_thor/README.md

+ 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

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


Loading…
Cancel
Save