| |
| seed_everything: 3407 |
| trainer: |
| logger: |
| class_path: pytorch_lightning.loggers.TensorBoardLogger |
| init_args: |
| save_dir: ./result/ |
| name: lightning_logs |
| version: null |
| log_graph: false |
| default_hp_metric: true |
| prefix: '' |
| sub_dir: null |
| logdir: null |
| comment: '' |
| purge_step: null |
| max_queue: 10 |
| flush_secs: 120 |
| filename_suffix: '' |
| write_to_disk: true |
| comet_config: |
| disabled: true |
| enable_checkpointing: true |
| callbacks: |
| - class_path: pytorch_lightning.callbacks.LearningRateMonitor |
| init_args: |
| logging_interval: null |
| log_momentum: false |
| - class_path: pytorch_lightning.callbacks.ModelSummary |
| init_args: |
| max_depth: 2 |
| - class_path: pytorch_lightning.callbacks.ModelCheckpoint |
| init_args: |
| dirpath: null |
| filename: wavtokenizer_checkpoint_{epoch}_{step}_{val_loss:.4f} |
| monitor: val_loss |
| verbose: false |
| save_last: true |
| save_top_k: 10 |
| save_weights_only: false |
| mode: min |
| auto_insert_metric_name: true |
| every_n_train_steps: 1000 |
| train_time_interval: null |
| every_n_epochs: null |
| save_on_train_epoch_end: null |
| - class_path: inspiremusic.wavtokenizer.decoder.helpers.GradNormCallback |
| default_root_dir: null |
| gradient_clip_val: null |
| gradient_clip_algorithm: null |
| num_nodes: 1 |
| num_processes: null |
| devices: -1 |
| gpus: null |
| auto_select_gpus: false |
| tpu_cores: null |
| ipus: null |
| enable_progress_bar: true |
| overfit_batches: 0.0 |
| track_grad_norm: -1 |
| check_val_every_n_epoch: 1 |
| fast_dev_run: false |
| accumulate_grad_batches: null |
| max_epochs: null |
| min_epochs: null |
| max_steps: 20000000 |
| min_steps: null |
| max_time: null |
| limit_train_batches: null |
| limit_val_batches: 100 |
| limit_test_batches: null |
| limit_predict_batches: null |
| val_check_interval: null |
| log_every_n_steps: 1000 |
| accelerator: gpu |
| strategy: ddp |
| sync_batchnorm: false |
| precision: 32 |
| enable_model_summary: true |
| num_sanity_val_steps: 2 |
| resume_from_checkpoint: null |
| profiler: null |
| benchmark: null |
| deterministic: null |
| reload_dataloaders_every_n_epochs: 0 |
| auto_lr_find: false |
| replace_sampler_ddp: true |
| detect_anomaly: false |
| auto_scale_batch_size: false |
| plugins: null |
| amp_backend: native |
| amp_level: null |
| move_metrics_to_cpu: false |
| multiple_trainloader_mode: max_size_cycle |
| inference_mode: true |
| ckpt_path: null |
| data: |
| class_path: inspiremusic.wavtokenizer.decoder.dataset.VocosDataModule |
| init_args: |
| train_params: |
| filelist_path: train.scp |
| sampling_rate: 24000 |
| num_samples: 72000 |
| batch_size: 38 |
| num_workers: 8 |
| val_params: |
| filelist_path: test.scp |
| sampling_rate: 24000 |
| num_samples: 72000 |
| batch_size: 10 |
| num_workers: 8 |
| model: |
| class_path: inspiremusic.wavtokenizer.decoder.experiment.WavTokenizer |
| init_args: |
| feature_extractor: |
| class_path: inspiremusic.wavtokenizer.decoder.feature_extractors.EncodecFeatures |
| init_args: |
| encodec_model: encodec_24khz |
| bandwidths: |
| - 6.6 |
| - 6.6 |
| - 6.6 |
| - 6.6 |
| train_codebooks: true |
| num_quantizers: 1 |
| dowmsamples: |
| - 8 |
| - 5 |
| - 4 |
| - 2 |
| vq_bins: 4096 |
| vq_kmeans: 200 |
| backbone: |
| class_path: inspiremusic.wavtokenizer.decoder.models.VocosBackbone |
| init_args: |
| input_channels: 512 |
| dim: 768 |
| intermediate_dim: 2304 |
| num_layers: 12 |
| layer_scale_init_value: null |
| adanorm_num_embeddings: 4 |
| head: |
| class_path: inspiremusic.wavtokenizer.decoder.heads.ISTFTHead |
| init_args: |
| dim: 768 |
| n_fft: 1280 |
| hop_length: 320 |
| padding: same |
| resume_config: config.yaml |
| resume_model: last.ckpt |
| sample_rate: 24000 |
| initial_learning_rate: 0.0001 |
| num_warmup_steps: 0 |
| mel_loss_coeff: 45.0 |
| mrd_loss_coeff: 1.0 |
| pretrain_mel_steps: 0 |
| decay_mel_coeff: false |
| evaluate_utmos: false |
| evaluate_pesq: true |
| evaluate_periodicty: true |
| resume: true |
|
|
