Skip to content

Commit

Permalink
Merge branch 'sparsity_int8_minorfix' into 'master'
Browse files Browse the repository at this point in the history
[SparsityINT8] Updated calib data size = 68 for better accuracy perf

See merge request adlsa/DL4AGX!3
  • Loading branch information
narendasan committed Jun 1, 2023
2 parents e3469c0 + a2b3e4a commit f963a04
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion SparsityINT8/step2_2_qat_training.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down

0 comments on commit f963a04

Please sign in to comment.