From e7f5b9eccc23137bd3260cc25ce1d3adec5bdeaa Mon Sep 17 00:00:00 2001 From: lbin Date: Tue, 17 Sep 2019 17:26:57 +0800 Subject: [PATCH] fixed print bug when using load_model args --- src/lib/models/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lib/models/model.py b/src/lib/models/model.py index fbc906e..643660e 100644 --- a/src/lib/models/model.py +++ b/src/lib/models/model.py @@ -54,7 +54,7 @@ def load_model(model, model_path, optimizer=None, resume=False, if state_dict[k].shape != model_state_dict[k].shape: print('Skip loading parameter {}, required shape{}, '\ 'loaded shape{}. {}'.format( - k, model_state_dict[k].shape, state_dict[k].shape) + msg) + k, model_state_dict[k].shape, state_dict[k].shape, msg)) state_dict[k] = model_state_dict[k] else: print('Drop parameter {}.'.format(k) + msg)