tc6/config.py

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
DEVICE = (
    "cuda"
    if torch.cuda.is_available()
    else "mps" if torch.backends.mps.is_available() else "cpu"
)