Aquila-Consortium/map2map
32
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix args helps

Browse Source
This commit is contained in:
Yin Li 2020-08-04 21:31:48 -04:00
parent 6b65b7888d
commit a1e5399311
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
map2map/args.py
View File

@ -37,9 +37,9 @@ def get_args():
def add_common_args(parser): def add_common_args(parser):
parser.add_argument('--in-norms', type=str_list, help='comma-sep. list ' parser.add_argument('--in-norms', type=str_list, help='comma-sep. list '
'of input normalization functions from .data.norms') 'of input normalization functions')
parser.add_argument('--tgt-norms', type=str_list, help='comma-sep. list ' parser.add_argument('--tgt-norms', type=str_list, help='comma-sep. list '
'of target normalization functions from .data.norms') 'of target normalization functions')
parser.add_argument('--crop', type=int, parser.add_argument('--crop', type=int,
help='size to crop the input and target data. Default is the ' help='size to crop the input and target data. Default is the '
'field size') 'field size')
@ -58,9 +58,9 @@ def add_common_args(parser):
'crop and pad are sizes of the input resolution') 'crop and pad are sizes of the input resolution')
parser.add_argument('--model', type=str, required=True, parser.add_argument('--model', type=str, required=True,
help='model from .models') help='(generator) model')
parser.add_argument('--criterion', default='MSELoss', type=str, parser.add_argument('--criterion', default='MSELoss', type=str,
help='model criterion from torch.nn') help='loss function')
parser.add_argument('--load-state', default=ckpt_link, type=str, parser.add_argument('--load-state', default=ckpt_link, type=str,
help='path to load the states of model, optimizer, rng, etc. ' help='path to load the states of model, optimizer, rng, etc. '
'Default is the checkpoint. ' 'Default is the checkpoint. '
@ -108,7 +108,7 @@ def add_train_args(parser):
'same factor for all fields') 'same factor for all fields')
parser.add_argument('--optimizer', default='Adam', type=str, parser.add_argument('--optimizer', default='Adam', type=str,
help='optimizer from torch.optim') help='optimization algorithm')
parser.add_argument('--lr', type=float, required=True, parser.add_argument('--lr', type=float, required=True,
help='initial learning rate') help='initial learning rate')
parser.add_argument('--optimizer-args', default='{}', type=json.loads, parser.add_argument('--optimizer-args', default='{}', type=json.loads,