Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

updated args

This commit is contained in:
William Falcon 2019-06-25 18:25:19 -04:00
parent 338f889e7c
commit a00b8f7861
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pytorch_lightning/utils/arg_parse.py
View File

@ -1,3 +1,5 @@
import pdb
def add_default_args(parser, root_dir, rand_seed=None, possible_model_names=None): def add_default_args(parser, root_dir, rand_seed=None, possible_model_names=None):
# tng, test, val check intervals # tng, test, val check intervals
@ -26,6 +28,7 @@ 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') parser.add_argument('--track_grad_norm', default=-1, type=int, help='if > 0, will track this grad norm')
# model saving # model saving
pdb.set_trace()
parser.add_argument('--model_save_path', default=root_dir + '/model_weights') 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_value', default='val_acc')
parser.add_argument('--model_save_monitor_mode', default='max') parser.add_argument('--model_save_monitor_mode', default='max')