dskill/DiffRhythm

1 2[DEFAULTS]3 4#name of the run5exp_name = F56 7# the batch size8batch_size = 8 9 10# the chunk size11max_frames = 3000 12min_frames = 1013 14# number of CPU workers for the DataLoader15num_workers = 416 17# the random seed18seed = 4219 20# Batches for gradient accumulation21accum_batches = 122 23# Number of steps between checkpoints24checkpoint_every = 10000                              25                     26# trainer checkpoint file to restart training from27ckpt_path = ''28 29# model checkpoint file to start a new training run from30pretrained_ckpt_path = ''31 32# Checkpoint path for the pretransform model if needed33pretransform_ckpt_path = ''34 35# configuration model specifying model hyperparameters36model_config = ''37 38# configuration for datasets39dataset_config = ''40 41# directory to save the checkpoints in42save_dir = ''43 44# grad norm45max_grad_norm = 1.046 47# grad accu48grad_accumulation_steps = 149 50# lr51learning_rate = 7.5e-552 53# epoch54epochs = 110 55 56# warmup steps57num_warmup_updates = 200058 59# save checkpoint per steps60save_per_updates = 500061 62# save last checkpoint per steps63last_per_steps = 500064 65prompt_path = "/mnt/sfs/music/lance/style-lance-full|/mnt/sfs/music/lance/style-lance-cnen-music-second"66lrc_path = "/mnt/sfs/music/lance/lrc-lance-emb-full|/mnt/sfs/music/lance/lrc-lance-cnen-second"67latent_path = "/mnt/sfs/music/lance/latent-lance|/mnt/sfs/music/lance/latent-lance-cnen-music-second-1|/mnt/sfs/music/lance/latent-lance-cnen-music-second-2"68 69audio_drop_prob = 0.370cond_drop_prob = 0.071style_drop_prob = 0.172lrc_drop_prob = 0.173 74align_lyrics = 075lyrics_slice = 076parse_lyrics = 177skip_empty_lyrics = 078lyrics_shift = -179 80use_style_prompt = 181 82tokenizer_type = gpt283 84reset_lr = 085 86resumable_with_seed = 66687 88downsample_rate = 204889 90grad_ckpt = 091 92dataset_path = "/mnt/sfs/music/hkchen/workspace/F5-TTS-HW/filelists/music123latent_asred_bpmstyle_cnen_pure1"93 94pure_prob = 0.0