| ########################################################### | |
| # DATA SETTING # | |
| ########################################################### | |
| data: | |
| path: "/dir/to/representations/" | |
| subset: | |
| train: "train_set_name" | |
| valid: "valid_set_name" | |
| test: "test_set_name" | |
| ########################################################### | |
| # MODEL SETTING # | |
| ########################################################### | |
| model_params: | |
| input_channels: 768 | |
| output_channels: 768 | |
| encode_channels: 768 | |
| decode_channels: 768 | |
| code_dim: 768 | |
| codebook_num: 1 | |
| codebook_size: 1024 | |
| bias: true | |
| enc_ratios: [1, 1] | |
| dec_ratios: [1, 1] | |
| enc_strides: [1, 1] # no downsampling | |
| dec_strides: [1, 1] | |
| enc_kernel_size: 3 | |
| dec_kernel_size: 3 | |
| enc_block_dilations: [1, 1] | |
| enc_block_kernel_size: 3 | |
| dec_block_dilations: [1, 1] | |
| dec_block_kernel_size: 3 | |
| ########################################################### | |
| # METRIC LOSS SETTING # | |
| ########################################################### | |
| repr_reconstruct_loss_params: | |
| loss_type: l2 | |
| ########################################################### | |
| # LOSS WEIGHT SETTING # | |
| ########################################################### | |
| lambda_vq_loss: 1.0 # Loss weight of vector quantize loss. | |
| lambda_repr_reconstruct_loss: 45.0 | |
| ########################################################### | |
| # DATA LOADER SETTING # | |
| ########################################################### | |
| batch_size: 32 # Batch size. | |
| batch_length: 96 # Length of each audio in batch (training w/o adv). | |
| pin_memory: true # Whether to pin memory in Pytorch DataLoader. | |
| num_workers: 4 # Number of workers in Pytorch DataLoader. | |
| ########################################################### | |
| # OPTIMIZER & SCHEDULER SETTING # | |
| ########################################################### | |
| model_optimizer_type: Adam | |
| model_optimizer_params: | |
| lr: 1.0e-4 | |
| betas: [0.5, 0.9] | |
| weight_decay: 0.0 | |
| model_scheduler_type: StepLR | |
| model_scheduler_params: | |
| step_size: 200000 # Model's scheduler step size. | |
| gamma: 1.0 | |
| grad_norm: -1 | |
| ########################################################### | |
| # INTERVAL SETTING # | |
| ########################################################### | |
| train_max_steps: 200000 # Number of training steps. (w/o adv) | |
| save_interval_steps: 20000 # Interval steps to save checkpoint. | |
| eval_interval_steps: 2000 # Interval steps to evaluate the network. | |
| log_interval_steps: 100 # Interval steps to record the training log. | |