From 1e556c2a0796b0351a21aa952d828aa0ae4056b6 Mon Sep 17 00:00:00 2001 From: wangmin Date: Mon, 12 Oct 2020 20:14:27 +0800 Subject: [PATCH] fix readmefile for resnet50_thor --- model_zoo/official/cv/resnet_thor/README.md | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) diff --git a/model_zoo/official/cv/resnet_thor/README.md b/model_zoo/official/cv/resnet_thor/README.md index 7598f427b0..85a52f11b0 100644 --- a/model_zoo/official/cv/resnet_thor/README.md +++ b/model_zoo/official/cv/resnet_thor/README.md @@ -97,7 +97,9 @@ sh run_eval_gpu.sh [DATASET_PATH] [CHECKPOINT_PATH] │ ├── thor_layer.py # thor layer │ └── dataset.py # data preprocessing ├── eval.py # infer script - └── train.py # train script + ├── train.py # train script + ├── export.py # export checkpoint file into air file + └── mindspore_hub_conf.py # config file for mindspore hub repository ``` ### Script Parameters @@ -228,20 +230,20 @@ Inference result will be stored in the example path, whose folder name is "eval" | Parameters | Ascend 910 | GPU | | -------------------------- | -------------------------------------- |---------------------------------- | | Model Version | ResNet50-v1.5 |ResNet50-v1.5| -| Resource | Ascend 910，CPU 2.60GHz 56cores，Memory 314G | GPU(Tesla V100 SXM2)，CPU 2.1GHz 24cores，Memory 128G -| uploaded Date | 06/01/2020 (month/day/year) | -| MindSpore Version | 0.3.0-alpha | | +| Resource | Ascend 910，CPU 2.60GHz 192cores，Memory 755G | GPU(Tesla V100 SXM2)，CPU 2.1GHz 24cores，Memory 128G +| uploaded Date | 06/01/2020 (month/day/year) | 09/23/2020(month/day/year) +| MindSpore Version | 0.3.0-alpha | 1.0.0 | | Dataset | ImageNet2012 | ImageNet2012| -| Training Parameters | epoch=42, steps per epoch=5004, batch_size = 32 |epoch=36, steps per epoch=5004, batch_size = 32 | +| Training Parameters | epoch=45, steps per epoch=5004, batch_size = 32 |epoch=40, steps per epoch=5004, batch_size = 32 | | Optimizer | THOR |THOR| | Loss Function | Softmax Cross Entropy |Softmax Cross Entropy | | outputs | probability | probability | | Loss |1.6453942 | 1.645802 | | Speed | 20.4ms/step（8pcs） |76ms/step（8pcs）| -| Total time | 72 mins | 229 mins| +| Total time(acc to 75.9%) | 72 mins | 229 mins| | Parameters (M) | 25.5 | 25.5 | | Checkpoint for Fine tuning | 491M (.ckpt file) |380M (.ckpt file) | -| Scripts | [Link](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/resnet_thor) |[Link](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/resnet_thor) | +| Scripts | [Link](https://gitee.com/mindspore/mindspore/tree/r1.0/model_zoo/official/cv/resnet_thor) |[Link](https://gitee.com/mindspore/mindspore/tree/r1.0/model_zoo/official/cv/resnet_thor) |