unity2009/MusicGenAI

1# @package __global__2 3defaults:4  - /solver/default5  - /model: score/basic6  - override /dset: audio/default7  - _self_8 9solver: diffusion10 11sample_rate: 1600012channels: 113compression_model_checkpoint: //sig/5091833e14n_q: 2   # number of codebooks to keep15 16dataset:17  batch_size: 818  num_workers: 1019  segment_duration: 120  train:21    num_samples: 10022  valid:23    num_samples: 10024  evaluate:25    batch_size: 826    num_samples: 1027  generate:28    batch_size: 829    num_samples: 1030    segment_duration: 1031 32loss:33  kind: mse34  norm_power: 0.35 36valid:37  every: 138 39evaluate:40  every: 541  num_workers: 542  metrics:43    visqol: false44    sisnr: false45    rvm: true46 47generate:48  every: 549  num_workers: 550  audio:51    sample_rate: ${sample_rate}52 53checkpoint:54  save_last: true55  save_every: 2556  keep_last: 1057  keep_every_states: null58 59 60optim:61  epochs: 5062  updates_per_epoch: 200063  lr: 2e-464  max_norm: 065  optimizer: adam66  adam:67    betas: [0.9, 0.999]68    weight_decay: 0.69  ema:70    use: true         # whether to use EMA or not71    updates: 1        # update at every step72    device: ${device} # device for EMA, can be put on GPU if more frequent updates73    decay: 0.99       # EMA decay value, if null, no EMA is used74 75processor:76  name: multi_band_processor77  use: false78  n_bands: 879  num_samples: 10_00080  power_std: 1.81 82resampling:83  use: false84  target_sr: 1600085 86filter:87  use: false88  n_bands: 489  idx_band: 090  cutoffs: null91 92schedule:93  repartition: "power"94  variable_step_batch: true95  beta_t0: 1.0e-596  beta_t1: 2.9e-297  beta_exp: 7.598  num_steps: 100099  variance: 'beta'100  clip: 5.101  rescale: 1.102  n_bands: null103  noise_scale: 1.0104 105metrics:106  num_stage: 4107