File size: 1,004 Bytes
5325fcc |
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 57 58 59 60 61 62 |
# @package __global__
# This is a minimal debugging configuration
# for MusicGen training solver
defaults:
- musicgen/default
- /model: lm/musicgen_lm
- override /model/lm/model_scale: xsmall
- override /dset: audio/example
- _self_
autocast: false
compression_model_checkpoint: //pretrained/debug_compression_model
transformer_lm:
n_q: 4
card: 400
conditioners:
description:
model: t5
t5:
name: t5-small
codebooks_pattern:
modeling: parallel
channels: 1
sample_rate: 32000
deadlock:
use: false # deadlock detection
dataset:
batch_size: 4
segment_duration: 5
sample_on_weight: false # Uniform sampling all the way
sample_on_duration: false # Uniform sampling all the way
generate:
audio:
strategy: peak
lm:
use_sampling: false
top_k: 0
top_p: 0.0
checkpoint:
save_every: 0
keep_last: 0
optim:
epochs: 2
updates_per_epoch: 10
optimizer: adamw
lr: 1e-4
logging:
log_tensorboard: true
schedule:
lr_scheduler: null
|