-
Notifications
You must be signed in to change notification settings - Fork 3
/
opt.py
37 lines (31 loc) · 1.79 KB
/
opt.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
32
33
34
35
36
37
import argparse
def get_opts():
parser = argparse.ArgumentParser()
parser.add_argument('--image_path', type=str, default='images/fox.jpg',
help='path to the image to reconstruct')
parser.add_argument('--img_wh', nargs="+", type=int, default=[800, 800],
help='resolution (img_w, img_h) of the image')
parser.add_argument('--use_pe', default=False, action='store_true',
help='use positional encoding or not')
parser.add_argument('--arch', type=str, default='identity',
choices=['relu', 'ff', 'siren', 'gabor', 'bacon',
'gaussian', 'quadratic', 'multi-quadratic',
'laplacian', 'super-gaussian', 'expsin'],
help='network structure')
parser.add_argument('--a', type=float, default=1.)
parser.add_argument('--b', type=float, default=1.)
parser.add_argument('--act_trainable', default=False, action='store_true',
help='whether to train activation hyperparameter')
parser.add_argument('--sc', type=float, default=10.,
help='fourier feature scale factor (std of the gaussian)')
parser.add_argument('--omega_0', type=float, default=30.,
help='omega in siren')
parser.add_argument('--batch_size', type=int, default=1024,
help='number of batch size')
parser.add_argument('--lr', type=float, default=1e-4,
help='learning rate')
parser.add_argument('--num_epochs', type=int, default=1000,
help='number of epochs')
parser.add_argument('--exp_name', type=str, default='exp',
help='experiment name')
return parser.parse_args()