forked from brody9512/PCG_FTSeg
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathconfig.py
More file actions
30 lines (22 loc) · 1.19 KB
/
config.py
File metadata and controls
30 lines (22 loc) · 1.19 KB
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
import argparse
def get_args():
parser = argparse.ArgumentParser(description='Configure specific Arguments')
parser.add_argument('--gpu', type=str, default='1')
parser.add_argument('--ver', type=int, default=45)
parser.add_argument('--featureLength', type=int, default=4096)
parser.add_argument('--target_sr', type=int, default=1000)
parser.add_argument('--lowpass', type=str, default=240)
parser.add_argument('--toler', type=int, default=40)
parser.add_argument('--year', type=int, default=2016)
parser.add_argument('--infer', action='store_true') # 2016
parser.add_argument('--infer_2022', action='store_true')
parser.add_argument('--nofolder', action='store_true')
parser.add_argument('--twice', action='store_true')
parser.add_argument('--conv_', action='store_true')
parser.add_argument('--fft', action='store_true')
parser.add_argument('--batch', type=int, default=64)
parser.add_argument('--not_2016', action='store_true')
parser.add_argument('--not_2022', action='store_true')
parser.add_argument('--not_amc', action='store_true')
parser.add_argument('--seed', type=int, default=42)
return parser.parse_args()