File size: 1,236 Bytes
aea7245 |
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 |
{
"train_dataset": "./data/train_dataset.csv",
"valid_dataset": "./data/eval_dataset.csv",
"train_duration": 0.38,
"valid_test_duration": 3.0,
"train_loader_length": null,
"batch_size": 32,
"num_workers": 16,
"num_epochs": 500,
"steps_per_sampler": 50000,
"learning_rate": 0.0001,
"encoder_dim": 64,
"encoder_rates": [
2,
4,
8,
8
],
"latent_dim": null,
"decoder_dim": 512,
"decoder_rates": [
8,
8,
4,
2
],
"n_codebooks": 4,
"codebook_size": 1024,
"codebook_dim": 8,
"quantizer_dropout": 0.5,
"sample_rate": 48000,
"commitment_loss_weight": 0.25,
"codebook_loss_weight": 1.0,
"l1loss": 1.0,
"stftloss": 0.0,
"melloss": 1.0,
"adam_beta1": 0.9,
"adam_beta2": 0.999,
"adam_weight_decay": 0.01,
"adam_epsilon": 1e-08,
"lr_scheduler_type": "cosine",
"num_warmup_steps": 0,
"gradient_accumulation_steps": 1,
"max_train_steps": 25000000,
"num_train_epochs": 500,
"device": "cuda",
"experiment_dir": "./exp",
"experiment_name": "mel1_wave1_no_adv",
"checkpoint_dir": "checkpoints",
"test_output_dir": "test_outputs"
} |