feature: 'FINETUNE_STRICT_LOAD' in config (#66)

cnstark · Sep 7, 2022 · ab481e4 · ab481e4
1 parent 2bc5c7c
commit ab481e4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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