small updates

2023-07-08 11:54:21 +02:00
parent c41a577672
commit 57b7aed4ba
15 changed files with 884 additions and 221 deletions
@@ -509,7 +509,10 @@ if __name__ == '__main__':

    step = 0
    if config['pretrained_model'] != '':
-        step = load_model(model, config['pretrained_model'], optimizer)
+        if 'ignore_saved_optimizer' in config.keys()  and config['ignore_saved_optimizer']:
+            step = load_model(model, config['pretrained_model'])
+        else:
+            step = load_model(model, config['pretrained_model'], optimizer=optimizer)

    if not args['debug']:
        # model = torch.compile(model)