| @@ -50,6 +50,13 @@ Dataset used: [CIFAR-10](<http://www.cs.toronto.edu/~kriz/cifar.html>) | |||||
| - Data format:binary files | - Data format:binary files | ||||
| - Note:Data will be processed in src/dataset.py | - 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) | # [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) | # [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 | '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) | ## [Training Process](#contents) | ||||
| @@ -297,6 +306,7 @@ Parameters for both training and evaluation can be set in config.py | |||||
| ### Evaluation Performance | ### Evaluation Performance | ||||
| #### GoogleNet on CIFAR-10 | |||||
| | Parameters | Ascend | GPU | | | Parameters | Ascend | GPU | | ||||
| | -------------------------- | ----------------------------------------------------------- | ---------------------- | | | -------------------------- | ----------------------------------------------------------- | ---------------------- | | ||||
| | Model Version | Inception V1 | Inception V1 | | | 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 | | | MindSpore Version | 0.7.0-alpha | 0.6.0-alpha | | ||||
| | Dataset | CIFAR-10 | CIFAR-10 | | | 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 | | | 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 | | | Loss Function | Softmax Cross Entropy | Softmax Cross Entropy | | ||||
| | outputs | probability | probobility | | | outputs | probability | probobility | | ||||
| | Loss | 0.0016 | 0.0016 | | | 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) | | | | 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) | | | 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 | ### Inference Performance | ||||
| #### GoogleNet on CIFAR-10 | |||||
| | Parameters | Ascend | GPU | | | Parameters | Ascend | GPU | | ||||
| | ------------------- | --------------------------- | --------------------------- | | | ------------------- | --------------------------- | --------------------------- | | ||||
| | Model Version | Inception V1 | Inception V1 | | | 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% | | | Accuracy | 1pc: 93.4%; 8pcs: 92.17% | 1pc: 93%, 8pcs: 92.89% | | ||||
| | Model for inference | 21.50M (.onnx file) | | | | 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) | ## [How to use](#contents) | ||||
| ### Inference | ### Inference | ||||