From a2b3e4a2b8fb082e6565d11e080cdcb74da8a9c6 Mon Sep 17 00:00:00 2001 From: Gwena Cunha Sergio Date: Thu, 1 Jun 2023 14:35:54 -0700 Subject: [PATCH] [SparsityINT8] Updated calib data size = 68 for better accuracy perf --- SparsityINT8/step2_2_qat_training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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,