From 775ca3736bb310d065a11c93b40fdd3b55a932b0 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Tue, 25 Jun 2019 18:29:16 -0400 Subject: [PATCH] updated args --- docs/source/examples/fully_featured_trainer.py | 4 ++-- pytorch_lightning/utils/arg_parse.py | 1 - 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/docs/source/examples/fully_featured_trainer.py b/docs/source/examples/fully_featured_trainer.py index e24da0af30..cb4e8ef004 100644 --- a/docs/source/examples/fully_featured_trainer.py +++ b/docs/source/examples/fully_featured_trainer.py @@ -158,8 +158,8 @@ if __name__ == '__main__': model_name = 'model_template' # use default args - pdb.set_trace() - root_dir = os.path.split(os.path.dirname(sys.modules['__main__'].__file__))[0] + root_dir = os.path.dirname(os.path.realpath(__file__)) + root_dir = os.path.join(root_dir, sys.modules['__main__'].__file__) parent_parser = get_default_parser(strategy='random_search', root_dir=root_dir) # allow model to overwrite or extend args diff --git a/pytorch_lightning/utils/arg_parse.py b/pytorch_lightning/utils/arg_parse.py index f762c98981..962ef78e4e 100644 --- a/pytorch_lightning/utils/arg_parse.py +++ b/pytorch_lightning/utils/arg_parse.py @@ -28,7 +28,6 @@ def add_default_args(parser, root_dir, rand_seed=None, possible_model_names=None parser.add_argument('--track_grad_norm', default=-1, type=int, help='if > 0, will track this grad norm') # model saving - pdb.set_trace() parser.add_argument('--model_save_path', default=root_dir + '/model_weights') parser.add_argument('--model_save_monitor_value', default='val_acc') parser.add_argument('--model_save_monitor_mode', default='max')