diff --git a/SparsityINT8/step2_2_qat_training.py b/SparsityINT8/step2_2_qat_training.py
index 9f371ad..e61cb0e 100644
--- a/SparsityINT8/step2_2_qat_training.py
+++ b/SparsityINT8/step2_2_qat_training.py
@@ -174,7 +174,7 @@ def train_qat(args, criterion, data_loader, data_loader_test, data_loader_val, d
     parser.add_argument("--test_data_size", type=int, default=None,
                         help="Dataset to be used for the final model evaluation (to obtain accuracy)."
                              " If None, take the entire val data. Otherwise, take subset.")
-    parser.add_argument("--calib_data_size", type=int, default=512,
+    parser.add_argument("--calib_data_size", type=int, default=68,
                         help="Dataset to be used for model calibration."
                              " If None, take the entire val data. Otherwise, take subset.")
     parser.add_argument("--val_data_size", type=int, default=None,