conf/vampnet-musdb-drums.yml

$include:
  - conf/vampnet.yml

VampNet.embedding_dim: 512
VampNet.n_layers: 12
VampNet.n_heads: 8

AudioDataset.duration: 12.0

train/AudioDataset.n_examples: 10000000
train/AudioLoader.sources:
  - /data/musdb18hq/train/**/*drums.wav


val/AudioDataset.n_examples: 500
val/AudioLoader.sources:
  - /data/musdb18hq/test/**/*drums.wav


test/AudioDataset.n_examples: 1000
test/AudioLoader.sources:
  - /data/musdb18hq/test/**/*drums.wav