From cc3bf30572cb791c2a96158f03ac59fc29d9821b Mon Sep 17 00:00:00 2001 From: magicse Date: Fri, 8 Dec 2023 12:58:32 +0200 Subject: [PATCH 1/2] Update train_first.py --- train_first.py | 1 + 1 file changed, 1 insertion(+) diff --git a/train_first.py b/train_first.py index 8a319d4..67dee8e 100644 --- a/train_first.py +++ b/train_first.py @@ -118,6 +118,7 @@ def main(config_path): if config.get('pretrained_model', '') != '': model, optimizer, start_epoch, iters = load_checkpoint(model, optimizer, config['pretrained_model'], load_only_params=config.get('load_only_params', True)) + start_epoch = start_epoch + 1 else: start_epoch = 0 iters = 0 From 60ef569b73bf455f684d3edc72d3a646e532afef Mon Sep 17 00:00:00 2001 From: magicse Date: Fri, 8 Dec 2023 13:03:58 +0200 Subject: [PATCH 2/2] Update train_second.py --- train_second.py | 1 + 1 file changed, 1 insertion(+) diff --git a/train_second.py b/train_second.py index 32fd30a..87560a1 100644 --- a/train_second.py +++ b/train_second.py @@ -132,6 +132,7 @@ def main(config_path): if config.get('pretrained_model', '') != '' and config.get('second_stage_load_pretrained', False): model, optimizer, start_epoch, iters = load_checkpoint(model, optimizer, config['pretrained_model'], load_only_params=config.get('load_only_params', True)) + start_epoch = start_epoch + 1 else: start_epoch = 0 iters = 0