提交 49426a06 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!4193 modify epoch for lenet-alexnet

Merge pull request !4193 from wukesong/modify_epoch
...@@ -22,7 +22,7 @@ alexnet_cfg = edict({ ...@@ -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,
......
...@@ -22,7 +22,7 @@ mnist_cfg = edict({ ...@@ -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,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册