{ "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" }