Skip to content
Snippets Groups Projects
Commit 2fb2d515 authored by Guy Jacob's avatar Guy Jacob
Browse files

compress_classifier: Bug fix in naming checkpoint created at quantized eval

parent 7596a0a6
No related branches found
No related tags found
No related merge requests found
......@@ -716,7 +716,7 @@ def evaluate_model(model, criterion, test_loader, loggers, activations_collector
if args.quantize_eval:
checkpoint_name = 'quantized'
apputils.save_checkpoint(0, args.arch, model, optimizer=None, best_top1=top1,
name='_'.split(args.name, checkpoint_name) if args.name else checkpoint_name,
name='_'.join([args.name, checkpoint_name]) if args.name else checkpoint_name,
dir=msglogger.logdir)
......
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