diff --git a/model_zoo/official/cv/alexnet/src/config.py b/model_zoo/official/cv/alexnet/src/config.py
index 9edfec2b60f8c98f9732e3209f3c7fd0973d67fa..65146795a9f7f0903e50fe5a0586f5ddd1c0a3c2 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 9a13ae535f219cae71004e334cf165a758cb962e..405b5ff5fa9d540248301d02ac5f262fa3e12f99 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,