From 74a401549a8f0f64bc409029b098ccc5c26c2e2c Mon Sep 17 00:00:00 2001
From: Ze Liu <liuze@mail.ustc.edu.cn>
Date: Wed, 14 Apr 2021 20:47:21 +0800
Subject: [PATCH] minor bug fixes

---
 main.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/main.py b/main.py
index eba7be3..18e9ad6 100644
--- 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')