diff --git a/official/cv/cnn_direction_model/README.md b/official/cv/cnn_direction_model/README.md index 1d2d989ea3cf9e60a419535882c4c75632504620..9b39d8774c2ae1c4fc83d795713b0879e6b563fe 100644 --- a/official/cv/cnn_direction_model/README.md +++ b/official/cv/cnn_direction_model/README.md @@ -292,17 +292,17 @@ top1_correct=8888, total=10202, acc=87.12% ### Evaluation Performance | Parameters | Ascend | -| -------------------------- | ------------------------------------------------------------| -| Resource | Ascend 910; CPU 2.60GHz, 192cores; Memory 755G; OS Euler2.8 | +| -------------------------- |-------------------------------------------------------------| +| Resource | Ascend 910; CPU 2.60GHz, 192cores; Memory 755G; OS Euler2.8 | | uploaded Date | 01/15/2021 (month/day/year) | -| MindSpore Version | 1.1 | -| Dataset | FSNS | -| Training Parameters | epoch=1, steps=104,477, batch_size = 20, lr=1e-07 | -| Optimizer | Adam | +| MindSpore Version | 1.1 | +| Dataset | FSNS | +| Training Parameters | epoch=5, steps=104,477, batch_size = 40, lr=5e-4 | +| Optimizer | Adam | | Loss Function | Softmax Cross Entropy | -| outputs | top 1 accuracy | -| Overall accuracy | 91.72% | -| Speed | 583 ms/step | +| outputs | top 1 accuracy | +| Overall accuracy | 91.72% | +| Speed | 583 ms/step | | Total time | 17 hours | # [Description of Random Situation](#contents) diff --git a/official/cv/cnn_direction_model/default_config.yaml b/official/cv/cnn_direction_model/default_config.yaml index b8ec741e7a2b422620885058082a4c007b890f57..2c0e3a4166757b707772febfc8acbd997fbf0114 100644 --- a/official/cv/cnn_direction_model/default_config.yaml +++ b/official/cv/cnn_direction_model/default_config.yaml @@ -28,7 +28,7 @@ mindrecord_dir: "" # training options dataset_name: "fsns" batch_size: 8 -epoch_size: 1 +epoch_size: 5 pretrain_epoch_size: 0 save_checkpoint: True save_checkpoint_steps: 2500