modify epoch

ae698843 · wukesong · a5850b95 · ae698843 · ae698843
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

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

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

未找到文件。
--- a/model_zoo/official/cv/alexnet/src/config.py
+++ b/model_zoo/official/cv/alexnet/src/config.py
@@ -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,

--- a/model_zoo/official/cv/lenet/src/config.py
+++ b/model_zoo/official/cv/lenet/src/config.py
@@ -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,