Huawei_Technology
/
mindspore
Not watched
1
0
Fork 0
Code
Releases 13 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
Browse Source

!4193 modify epoch for lenet-alexnet 

Merge pull request !4193 from wukesong/modify_epoch
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
13a66805b3 ae698843c7
commit
49426a0639
2 changed files with 2 additions and 2 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -1
      model_zoo/official/cv/alexnet/src/config.py
  2. +1
    -1
      model_zoo/official/cv/lenet/src/config.py

+ 1
- 1
model_zoo/official/cv/alexnet/src/config.py View File

@@ -22,7 +22,7 @@ alexnet_cfg = edict({
'num_classes': 10,
'learning_rate': 0.002,
'momentum': 0.9,
'epoch_size': 1,
'epoch_size': 30,
'batch_size': 32,
'buffer_size': 1000,
'image_height': 227,


+ 1
- 1
model_zoo/official/cv/lenet/src/config.py View File

@@ -22,7 +22,7 @@ mnist_cfg = edict({
'num_classes': 10,
'lr': 0.01,
'momentum': 0.9,
'epoch_size': 1,
'epoch_size': 10,
'batch_size': 32,
'buffer_size': 1000,
'image_height': 32,


Write Preview
Loading…
Cancel
Save