From 467b3d210786a794eaecba99b74cedd44390570c Mon Sep 17 00:00:00 2001 From: YuAng Date: Wed, 16 Jun 2021 14:48:23 +0800 Subject: [PATCH] Fix train.py parse_args bug --- train.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/train.py b/train.py index fb7b94f..fa8a0bd 100644 --- a/train.py +++ b/train.py @@ -25,7 +25,9 @@ def parse_args(): # check documentation: https://pytorch-lightning.readthedocs.io/en/latest/common/trainer.html#trainer-flags parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter) parser.add_argument( + 'data_cfg_path', type=str, help='data config path') parser.add_argument( + 'main_cfg_path', type=str, help='main config path') parser.add_argument( '--exp_name', type=str, default='default_exp_name') parser.add_argument( @@ -37,6 +39,7 @@ def parse_args(): nargs='?', default=True, help='whether loading data to pinned memory or not') parser.add_argument( '--ckpt_path', type=str, default=None, + help='pretrained checkpoint path, helpful for using a pre-trained coarse-only LoFTR') parser.add_argument( '--disable_ckpt', action='store_true', help='disable checkpoint saving (useful for debugging).') @@ -46,7 +49,7 @@ def parse_args(): parser.add_argument( '--parallel_load_data', action='store_true', help='load datasets in with multiple processes.') - + parser = pl.Trainer.add_argparse_args(parser) return parser.parse_args()