-
Notifications
You must be signed in to change notification settings - Fork 13
/
util.py
31 lines (28 loc) · 1.58 KB
/
util.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
import os
from argparse import ArgumentParser
def get_args():
parser = ArgumentParser(description='grass_pytorch')
parser.add_argument('--box_code_size', type=int, default=12)
parser.add_argument('--feature_size', type=int, default=80)
parser.add_argument('--hidden_size', type=int, default=200)
parser.add_argument('--symmetry_size', type=int, default=8)
parser.add_argument('--max_box_num', type=int, default=30)
parser.add_argument('--max_sym_num', type=int, default=10)
parser.add_argument('--epochs', type=int, default=300)
parser.add_argument('--batch_size', type=int, default=123)
parser.add_argument('--show_log_every', type=int, default=3)
parser.add_argument('--save_log', action='store_true', default=False)
parser.add_argument('--save_log_every', type=int, default=3)
parser.add_argument('--save_snapshot', action='store_true', default=False)
parser.add_argument('--save_snapshot_every', type=int, default=5)
parser.add_argument('--no_plot', action='store_true', default=False)
parser.add_argument('--lr', type=float, default=.001)
parser.add_argument('--lr_decay_by', type=float, default=1)
parser.add_argument('--lr_decay_every', type=float, default=1)
parser.add_argument('--no_cuda', action='store_true', default=False)
parser.add_argument('--gpu', type=int, default=0)
parser.add_argument('--data_path', type=str, default='data')
parser.add_argument('--save_path', type=str, default='models')
parser.add_argument('--resume_snapshot', type=str, default='')
args = parser.parse_args()
return args