| exp: |
| dir: ./Exps |
| name: ApolloVoice |
| model_name: ApolloVoice |
| datas: |
| _target_: look2hear.datas.MusdbMoisesdbDataModule |
| train_dir: ./dataset/restoration/train |
| eval_dir: ./dataset/restoration/test |
| codec_type: mp3 |
| codec_options: |
| bitrate: random |
| compression: random |
| complexity: random |
| vbr: random |
| sr: 44100 |
| segments: 5.4 |
| num_stems: 8 |
| snr_range: |
| - -10 |
| - 10 |
| num_samples: 3000 |
| batch_size: 1 |
| num_workers: 8 |
| model: |
| _target_: look2hear.models.apollo.Apollo |
| sr: 44100 |
| win: 20 |
| feature_dim: 384 |
| layer: 8 |
| discriminator: |
| _target_: look2hear.discriminators.frequencydis.MultiFrequencyDiscriminator |
| nch: 2 |
| window: |
| - 32 |
| - 64 |
| - 128 |
| - 256 |
| - 512 |
| - 1024 |
| - 2048 |
| optimizer_g: |
| _target_: bitsandbytes.optim.AdamW8bit |
| lr: 0.001 |
| weight_decay: 0.01 |
| optimizer_d: |
| _target_: bitsandbytes.optim.AdamW8bit |
| lr: 0.0001 |
| weight_decay: 0.01 |
| betas: |
| - 0.5 |
| - 0.99 |
| scheduler_g: |
| _target_: torch.optim.lr_scheduler.StepLR |
| step_size: 4 |
| gamma: 0.98 |
| scheduler_d: |
| _target_: torch.optim.lr_scheduler.StepLR |
| step_size: 4 |
| gamma: 0.98 |
| loss_g: |
| _target_: look2hear.losses.gan_losses.MultiFrequencyGenLoss |
| eps: 2.0e-08 |
| loss_d: |
| _target_: look2hear.losses.gan_losses.MultiFrequencyDisLoss |
| eps: 2.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: 5 |
| 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: true |
| project: Audio-Restoration |
| trainer: |
| _target_: pytorch_lightning.Trainer |
| devices: |
| - 0 |
| max_epochs: 500 |
| sync_batchnorm: true |
| default_root_dir: ${exp.dir}/${exp.name}/ |
| accelerator: cuda |
| limit_train_batches: 1.0 |
| fast_dev_run: false |
|
|
