diff --git a/examples/tensorflow/classification/configs/quantization/mobilenet_v2_imagenet_int8.json b/examples/tensorflow/classification/configs/quantization/mobilenet_v2_imagenet_int8.json
index e18678ba443..1a73fd784d4 100644
--- a/examples/tensorflow/classification/configs/quantization/mobilenet_v2_imagenet_int8.json
+++ b/examples/tensorflow/classification/configs/quantization/mobilenet_v2_imagenet_int8.json
@@ -6,7 +6,7 @@
     },
 
     "batch_size": 256,
-    "epochs": 18,
+    "epochs": 16,
     "num_classes": 1001,
     "dataset_preprocessing_preset": "imagenet2012_slim",
 
@@ -14,8 +14,8 @@
         "type": "Adam",
         "schedule_type": "piecewise_constant",
         "schedule_params": {
-            "boundaries": [8, 12, 15],
-            "values": [5e-4, 5e-5, 5e-6, 1e-6]
+            "boundaries": [6, 10],
+            "values": [1e-4, 1e-5, 1e-6]
         }
     },
 
diff --git a/examples/tensorflow/classification/main.py b/examples/tensorflow/classification/main.py
index cc069444602..89fecd7a064 100644
--- a/examples/tensorflow/classification/main.py
+++ b/examples/tensorflow/classification/main.py
@@ -306,8 +306,8 @@ def run(config):
         logger.info('training...')
         compress_model.fit(
             train_dataset,
-            epochs=1,#train_epochs,
-            steps_per_epoch=3,#train_steps,
+            epochs=train_epochs,
+            steps_per_epoch=train_steps,
             initial_epoch=initial_epoch,
             callbacks=callbacks,
             **validation_kwargs)