File size: 769 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 |
# @package __global__
defaults:
- /solver/musicgen/default
- _self_
- /solver/audiogen/evaluation: none
- override /dset: audio/default
# See config/solver/musicgen/default.yaml for a list of possible values.
# We only keep the most important here.
autocast: true
autocast_dtype: float16
solver: audiogen
sample_rate: ???
channels: ???
compression_model_checkpoint: ???
tokens:
padding_with_special_token: false
dataset:
batch_size: 128
segment_duration: 10
min_segment_ratio: 1.0 # lower values such as 0.5 result in generations with a lot of silence.
optim:
epochs: 100
updates_per_epoch: 2000
lr: 1e-4
optimizer: adamw
max_norm: 1.0
adam:
betas: [0.9, 0.95]
weight_decay: 0.1
eps: 1e-8
schedule:
lr_scheduler: null
|