From ab481e4ac7fa215bb6893aff9544ab93b6ab128d Mon Sep 17 00:00:00 2001
From: Yuhao Wang <45590791+cnstark@users.noreply.github.com>
Date: Wed, 7 Sep 2022 14:35:25 +0800
Subject: [PATCH] feature: 'FINETUNE_STRICT_LOAD' in config (#66)

---
 easytorch/core/runner.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/easytorch/core/runner.py b/easytorch/core/runner.py
index 226984e..87f05dc 100644
--- a/easytorch/core/runner.py
+++ b/easytorch/core/runner.py
@@ -408,7 +408,7 @@ def init_training(self, cfg: Dict):
 
         # fine tune
         if hasattr(cfg['TRAIN'], 'FINETUNE_FROM'):
-            self.load_model(cfg['TRAIN']['FINETUNE_FROM'])
+            self.load_model(cfg['TRAIN']['FINETUNE_FROM'], cfg['TRAIN'].get('FINETUNE_STRICT_LOAD', True))
             self.logger.info('Start fine tuning')
 
         # resume