Spaces:
Running
on
Zero
Running
on
Zero
File size: 638 Bytes
812b01c |
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 |
import torch
# ─── 1) CONFIG ─────────────────────────────────────────────────────
SAMPLE_RATE = 22050
N_MELS = 80
HOP_LENGTH = 256
TIME_SUB = 1
CNN_CH = 256
N_HEADS = 8
D_MODEL = 512
FF_DIM = 1024
N_LAYERS = 6
DEPTHWISE_CONV_KERNEL_SIZE = 31
DROPOUT = 0.1
HIDDEN_DIM = 64
N_TYPES = 7
BATCH_SIZE = 2
GRAD_ACCUM_STEPS = 8
LR = 3e-4
EPOCHS = 200
NPS_PENALTY_WEIGHT_ALPHA = 0.3
NPS_PENALTY_WEIGHT_BETA = 1.0
DEVICE = (
"cuda"
if torch.cuda.is_available()
else "mps" if torch.backends.mps.is_available() else "cpu"
)
|