unet_sr_baseline/metrics/config.json

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