Skip to content

Commit be9d901

Browse files
committed
2 parents 240f0a7 + b3d6c4f commit be9d901

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

lib/scene_parser/rcnn/utils/checkpoint.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -64,13 +64,13 @@ def load(self, f=None, resume=0, use_latest=False):
6464
self.logger.info("Loading checkpoint from {}".format(f))
6565
checkpoint = self._load_file(f)
6666
self._load_model(checkpoint)
67-
if "optimizer" in checkpoint and self.optimizer and "sg" in f and not self.inference:
68-
self.logger.info("Loading optimizer from {}".format(f))
69-
self.optimizer.load_state_dict(checkpoint.pop("optimizer"))
70-
if "scheduler" in checkpoint and self.scheduler and "sg" in f and not self.inference:
71-
self.logger.info("Loading scheduler from {}".format(f))
72-
self.scheduler.load_state_dict(checkpoint.pop("scheduler"))
73-
checkpoint['iteration'] = resume # if we load detector, the we should not use its start iteration
67+
# if "optimizer" in checkpoint and self.optimizer and "sg" in f and not self.inference:
68+
# self.logger.info("Loading optimizer from {}".format(f))
69+
# self.optimizer.load_state_dict(checkpoint.pop("optimizer"))
70+
# if "scheduler" in checkpoint and self.scheduler and "sg" in f and not self.inference:
71+
# self.logger.info("Loading scheduler from {}".format(f))
72+
# self.scheduler.load_state_dict(checkpoint.pop("scheduler"))
73+
# checkpoint['iteration'] = resume # if we load detector, the we should not use its start iteration
7474

7575
return checkpoint
7676

0 commit comments

Comments
 (0)