diff --git a/trainers/base.py b/trainers/base.py index 1713541..2b7035c 100644 --- a/trainers/base.py +++ b/trainers/base.py @@ -36,7 +36,7 @@ def __init__(self, cfg, model, dataset): model.load_state_dict(infos['state_dict']) self.start_epoch = infos['epoch'] self.best_rank1 = infos['best_rank1'] - self.logger.write(' loaded\n'%self.resume) + self.logger.write(' loaded %s\n'%self.resume) else: self.start_epoch = 0 self.best_rank1 = 0