diff --git a/model_zoo/official/cv/alexnet/src/config.py b/model_zoo/official/cv/alexnet/src/config.py
index 9edfec2b60..65146795a9 100644
--- 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,
diff --git a/model_zoo/official/cv/lenet/src/config.py b/model_zoo/official/cv/lenet/src/config.py
index 9a13ae535f..405b5ff5fa 100644
--- 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,