p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

modify epoch

This commit is contained in:
wukesong 2020-08-10 14:43:46 +08:00
parent a5850b9551
commit ae698843c7
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model_zoo/official/cv/alexnet/src/config.py
View File

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

2
model_zoo/official/cv/lenet/src/config.py
View File

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