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