File size: 464 Bytes
05d6e12
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
import torch


SAMPLE_RATE = 16000
HOP_LENGTH = 512
ONSET_LENGTH = HOP_LENGTH
OFFSET_LENGTH = HOP_LENGTH

HOPS_IN_ONSET = ONSET_LENGTH // HOP_LENGTH
HOPS_IN_OFFSET = OFFSET_LENGTH // HOP_LENGTH
MIN_MIDI = 21
MAX_MIDI = 108
N_KEYS = MAX_MIDI - MIN_MIDI + 1

DTW_FACTOR = 3

N_MELS = 229
MEL_FMIN = 30
MEL_FMAX = SAMPLE_RATE // 2
WINDOW_LENGTH = 2048

SEQ_LEN = 327680  # 20 seconds

DRUM_CHANNEL = 9

DEFAULT_DEVICE = "cuda" if torch.cuda.is_available() else "cpu"