minor bug fixes

2021-04-14 20:47:21 +08:00 · 2021-04-14 20:47:21 +08:00 · 74a401549a
commit 74a401549a
parent c7e6d6efbd
1 changed files with 2 additions and 0 deletions
--- a/main.py
+++ b/main.py
@ -110,7 +110,9 @@ def main(config):
        if resume_file:
            if config.MODEL.RESUME:
                logger.warning(f"auto-resume changing resume file from {config.MODEL.RESUME} to {resume_file}")
+            config.defrost()
            config.MODEL.RESUME = resume_file
+            config.freeze()
            logger.info(f'auto resuming from {resume_file}')
        else:
            logger.info(f'no checkpoint found in {config.OUTPUT}, ignoring auto resume')