-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfig.py
57 lines (36 loc) · 998 Bytes
/
config.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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
## data params
## pre process
beat_resolution = 4
combine_instrus = True
min_octave = 3
max_octave = 5
## post process
multi_octave = True
polyphony = False
monophony_mode = 'h' # 'l':lowest, 'h':highest, 'm':most
data_path = 'data'
dev_ratio = 0
## model params
timestep_size = 12+1 if not multi_octave else 12*(max_octave-min_octave+1)+1
in_size = timestep_size
state_size = 2**5
out_size = timestep_size
## train params
learning_rate = 1e-1
max_seq_len = 20
batch_size = 0
hm_epochs = 1
hm_epochs_per_t = 1000
optimizer = 'adaptive'
model_path = 'models/model'
fresh_model = True
fresh_meta = True
use_gpu = False
## interact params
pick_threshold = 1/((max_octave-min_octave+1)*12+1) if multi_octave else 1/13
hm_extra_steps = beat_resolution*4 *2
hm_output_file = 1
##
config_to_save = ['beat_resolution', 'min_octave', 'max_octave', 'multi_octave', 'polyphony',
'timestep_size', 'in_size', 'out_size', 'state_size', 'act_fn',
]