| | |
| | 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 |
| |
|
