[SparsityINT8] Updated calib data size = 68 for better accuracy perf

NVIDIA · Jun 1, 2023 · a2b3e4a · a2b3e4a
1 parent e3469c0
commit a2b3e4a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,