From 73df57bf08c0d29cde9c19b3feb38d1a7a37713e Mon Sep 17 00:00:00 2001
From: Neta Zmora <neta.zmora@intel.com>
Date: Fri, 29 Jun 2018 21:49:27 +0300
Subject: [PATCH] Bug fix: compression schedule configuration parsing
Used the wrong indentation when parsing RegularizationPolicy
---
apputils/checkpoint.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/apputils/checkpoint.py b/apputils/checkpoint.py
index 16c9c98..69e2a7c 100755
--- a/apputils/checkpoint.py
+++ b/apputils/checkpoint.py
@@ -43,17 +43,17 @@ def save_checkpoint(epoch, arch, model, optimizer=None, scheduler=None,
name: the name of the checkpoint file
dir: directory in which to save the checkpoint
"""
- msglogger.info("Saving checkpoint")
if not os.path.isdir(dir):
msglogger.info("Error: Directory to save checkpoint doesn't exist - {0}".format(os.path.abspath(dir)))
exit(1)
filename = 'checkpoint.pth.tar' if name is None else name + '_checkpoint.pth.tar'
fullpath = os.path.join(dir, filename)
+ msglogger.info("Saving checkpoint to: %s" % fullpath)
filename_best = 'best.pth.tar' if name is None else name + '_best.pth.tar'
fullpath_best = os.path.join(dir, filename_best)
checkpoint = {}
checkpoint['epoch'] = epoch
- checkpoint['arch'] = arch
+ checkpoint['arch'] = arch
checkpoint['state_dict'] = model.state_dict()
if best_top1 is not None:
checkpoint['best_top1'] = best_top1
@@ -85,6 +85,7 @@ def load_checkpoint(model, chkpt_file, optimizer=None):
if os.path.isfile(chkpt_file):
msglogger.info("=> loading checkpoint %s", chkpt_file)
checkpoint = torch.load(chkpt_file)
+ msglogger.info("Checkpoint keys:\n{}".format("\n\t".join(k for k in checkpoint.keys())))
start_epoch = checkpoint['epoch'] + 1
best_top1 = checkpoint.get('best_top1', None)
if best_top1 is not None:
--
GitLab