| # @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 | |
| 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 | |