|
|
exp: |
|
|
dir: ./Exps |
|
|
name: Apollo |
|
|
|
|
|
|
|
|
|
|
|
datas: |
|
|
_target_: look2hear.datas.MusdbMoisesdbDataModule |
|
|
train_dir: ./hdf5_datas |
|
|
eval_dir: ./eval |
|
|
codec_type: mp3 |
|
|
codec_options: |
|
|
bitrate: random |
|
|
compression: random |
|
|
complexity: random |
|
|
vbr: random |
|
|
sr: 44100 |
|
|
segments: 3 |
|
|
num_stems: 8 |
|
|
snr_range: [-10, 10] |
|
|
num_samples: 40000 |
|
|
batch_size: 1 |
|
|
num_workers: 8 |
|
|
|
|
|
model: |
|
|
|
|
|
sr: 44100 |
|
|
win: 20 |
|
|
feature_dim: 256 |
|
|
layer: 6 |
|
|
|
|
|
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: 1e-8 |
|
|
|
|
|
loss_d: |
|
|
_target_: look2hear.losses.gan_losses.MultiFrequencyDisLoss |
|
|
eps: 1e-8 |
|
|
|
|
|
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: 20 |
|
|
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: false |
|
|
project: Audio-Restoration |
|
|
|
|
|
trainer: |
|
|
_target_: pytorch_lightning.Trainer |
|
|
devices: [0,1,2,3,4,5,6,7] |
|
|
max_epochs: 500 |
|
|
sync_batchnorm: true |
|
|
default_root_dir: ${exp.dir}/${exp.name}/ |
|
|
accelerator: cuda |
|
|
limit_train_batches: 1.0 |
|
|
fast_dev_run: false |
|
|
|
