mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-30 13:40:46 +08:00
updated args
This commit is contained in:
1 parent
338f889e7c
commit
a00b8f7861
1 file changed
+3
@@ -1,3 +1,5 @@
|
||||
import pdb
|
||||
|
||||
def add_default_args(parser, root_dir, rand_seed=None, possible_model_names=None):
|
||||
|
||||
# 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')
|
||||
|
||||
# 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')
|
||||
|
||||
Reference in new issue
Block a user