Skip to content
Snippets Groups Projects
Commit 3df06fb4 authored by SunYiran's avatar SunYiran Committed by Guy Jacob
Browse files

Bugfix - Save stats to file when 'qe-calibration' arg is used (#437)

parent 8e3f04cc
No related branches found
No related tags found
No related merge requests found
......@@ -99,7 +99,7 @@ def handle_subapps(model, criterion, optimizer, compression_scheduler, pylogger,
args.dataset, add_softmax=True, verbose=False)
do_exit = True
elif args.qe_calibration and not (args.evaluate and args.quantize_eval):
classifier.acts_quant_stats_collection(model, criterion, pylogger, args)
classifier.acts_quant_stats_collection(model, criterion, pylogger, args, save_to_file=True)
do_exit = True
elif args.activation_histograms:
classifier.acts_histogram_collection(model, criterion, pylogger, args)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment