diff --git a/model_zoo/official/cv/googlenet/README.md b/model_zoo/official/cv/googlenet/README.md index 422ebd676a..ac7aeb7ab4 100644 --- a/model_zoo/official/cv/googlenet/README.md +++ b/model_zoo/official/cv/googlenet/README.md @@ -50,6 +50,13 @@ Dataset used: [CIFAR-10]() - Data format:binary files - Note:Data will be processed in src/dataset.py +Dataset used can refer to paper. + +- Dataset size: 125G, 1250k colorful images in 1000 classes + - Train: 120G, 1200k images + - Test: 5G, 50k images +- Data format: RGB images. + - Note: Data will be processed in src/dataset.py # [Features](#contents) @@ -116,6 +123,7 @@ After installing MindSpore via the official website, you can start training and ``` +We use CIFAR-10 dataset by default. Your can also pass `$dataset_type` to the scripts so that select different datasets. For more details, please refer the specify script. # [Script Description](#contents) @@ -167,6 +175,7 @@ Parameters for both training and evaluation can be set in config.py 'geir_filename': 'googlenet.geir' # file name of the geir model used in export.py ``` +For more configuration details, please refer the script `config.py`. ## [Training Process](#contents) @@ -297,6 +306,7 @@ Parameters for both training and evaluation can be set in config.py ### Evaluation Performance +#### GoogleNet on CIFAR-10 | Parameters | Ascend | GPU | | -------------------------- | ----------------------------------------------------------- | ---------------------- | | Model Version | Inception V1 | Inception V1 | @@ -305,7 +315,7 @@ Parameters for both training and evaluation can be set in config.py | MindSpore Version | 0.7.0-alpha | 0.6.0-alpha | | Dataset | CIFAR-10 | CIFAR-10 | | Training Parameters | epoch=125, steps=390, batch_size = 128, lr=0.1 | epoch=125, steps=390, batch_size=128, lr=0.1 | -| Optimizer | SGD | SGD | +| Optimizer | Momentum | Momentum | | Loss Function | Softmax Cross Entropy | Softmax Cross Entropy | | outputs | probability | probobility | | Loss | 0.0016 | 0.0016 | @@ -316,9 +326,29 @@ Parameters for both training and evaluation can be set in config.py | Model for inference | 21.50M (.onnx file), 21.60M(.air file) | | | Scripts | [googlenet script](https://gitee.com/mindspore/mindspore/tree/r0.7/model_zoo/official/cv/googlenet) | [googlenet script](https://gitee.com/mindspore/mindspore/tree/r0.6/model_zoo/official/cv/googlenet) | +#### GoogleNet on 1200k images +| Parameters | Ascend | +| -------------------------- | ----------------------------------------------------------- | +| Model Version | Inception V1 | +| Resource | Ascend 910, CPU 2.60GHz, 56cores, Memory 314G | +| uploaded Date | 09/20/2020 (month/day/year) | +| MindSpore Version | 0.7.0-alpha | +| Dataset | 1200k images | +| Training Parameters | epoch=300, steps=5000, batch_size=256, lr=0.1 | +| Optimizer | Momentum | +| Loss Function | Softmax Cross Entropy | +| outputs | probability | +| Loss | 2.0 | +| Speed | 1pc: 152 ms/step; 8pcs: 171 ms/step | +| Total time | 8pcs: 8.8 hours | +| Parameters (M) | 13.0 | +| Checkpoint for Fine tuning | 52M (.ckpt file) | +| Scripts | [googlenet script](https://gitee.com/mindspore/mindspore/tree/r0.7/model_zoo/official/cv/googlenet) | + ### Inference Performance +#### GoogleNet on CIFAR-10 | Parameters | Ascend | GPU | | ------------------- | --------------------------- | --------------------------- | | Model Version | Inception V1 | Inception V1 | @@ -331,6 +361,18 @@ Parameters for both training and evaluation can be set in config.py | Accuracy | 1pc: 93.4%; 8pcs: 92.17% | 1pc: 93%, 8pcs: 92.89% | | Model for inference | 21.50M (.onnx file) | | +#### GoogleNet on 1200k images +| Parameters | Ascend | +| ------------------- | --------------------------- | +| Model Version | Inception V1 | +| Resource | Ascend 910 | +| Uploaded Date | 09/20/2020 (month/day/year) | +| MindSpore Version | 0.7.0-alpha | +| Dataset | 1200k images | +| batch_size | 256 | +| outputs | probability | +| Accuracy | 8pcs: 71.81% | + ## [How to use](#contents) ### Inference