| exp: |
| dir: ./exps |
| name: oggrestore |
| datas: |
| _target_: look2hear.datas.DataModule |
| dataset_type: 1 |
| sr: 44100 |
| segments: 4 |
| num_steps: 1000 |
| batch_size: 1 |
| num_workers: 0 |
| pin_memory: true |
| stems: |
| original: original |
| codec: codec |
| train: |
| dir: |
| - train |
| original_format: wav |
| codec_format: ogg |
| valid: |
| dir: |
| - vaild |
| original_format: wav |
| codec_format: ogg |
| model: |
| _target_: look2hear.models.apollo.Apollo |
| sr: 44100 |
| win: 20 |
| feature_dim: 64 |
| layer: 4 |
| discriminator: |
| _target_: look2hear.discriminators.frequencydis.MultiFrequencyDiscriminator |
| nch: 2 |
| window: |
| - 32 |
| - 64 |
| - 128 |
| - 256 |
| - 512 |
| - 1024 |
| - 2048 |
| optimizer_g: |
| _target_: torch.optim.AdamW |
| lr: 0.001 |
| weight_decay: 0.01 |
| optimizer_d: |
| _target_: torch.optim.AdamW |
| lr: 0.0001 |
| weight_decay: 0.01 |
| betas: |
| - 0.5 |
| - 0.99 |
| scheduler_g: |
| _target_: torch.optim.lr_scheduler.StepLR |
| step_size: 2 |
| gamma: 0.98 |
| scheduler_d: |
| _target_: torch.optim.lr_scheduler.StepLR |
| step_size: 2 |
| gamma: 0.98 |
| loss_g: |
| _target_: look2hear.losses.gan_losses.MultiFrequencyGenLoss |
| eps: 1.0e-08 |
| loss_d: |
| _target_: look2hear.losses.gan_losses.MultiFrequencyDisLoss |
| eps: 1.0e-08 |
| metrics: |
| _target_: look2hear.losses.MultiSrcNegSDR |
| sdr_type: sisdr |
| system: |
| _target_: look2hear.system.audio_litmodule.AudioLightningModule |
| early_stopping: |
| _target_: pytorch_lightning.callbacks.EarlyStopping |
| monitor: val_loss |
| patience: 50 |
| mode: min |
| verbose: true |
| checkpoint: |
| _target_: pytorch_lightning.callbacks.ModelCheckpoint |
| dirpath: ${exp.dir}/${exp.name}/checkpoints |
| monitor: val_loss |
| mode: min |
| verbose: true |
| save_top_k: 10 |
| save_last: true |
| filename: '{epoch}-{val_loss:.4f}' |
| logger: |
| _target_: pytorch_lightning.loggers.WandbLogger |
| name: ${exp.name} |
| save_dir: ${exp.dir}/${exp.name}/logs |
| offline: false |
| project: Audio-Restoration |
| trainer: |
| _target_: pytorch_lightning.Trainer |
| devices: |
| - 0 |
| max_epochs: 50 |
| sync_batchnorm: true |
| default_root_dir: ${exp.dir}/${exp.name}/ |
| accelerator: cuda |
| limit_train_batches: 1.0 |
| fast_dev_run: false |
| precision: bf16 |
| enable_model_summary: true |
|
|
