File size: 1,017 Bytes
59b7eeb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
trainer:
accelerator: 'gpu'
num_nodes: 1
devices: 1
min_steps: 3_000_000
max_steps: 3_000_000
precision: 16-mixed
limit_val_batches: 2560
val_check_interval: 4000
num_sanity_val_steps: 0
accumulate_grad_batches: 1
# loss
lambdas:
lambda_disc: 1.0
lambda_feat_match_loss: 1.0
lambda_mel_loss: 15.0
lambda_adv: 1.0
lambda_stft_loss: 1.0
lambda_semantic_loss: 5
lambda_perceptual_loss: 0
use_mel_loss: true
use_feat_match_loss: true
use_stft_loss: false
stft_loss_params:
fft_sizes: [512,2048]
hop_sizes: [128,512]
win_lengths: [512,2048]
window: hann_window
# optimizer
gen_optim_params:
lr: 1.0
betas: [0.8, 0.9]
disc_optim_params:
lr: 1.0
betas: [0.8, 0.9]
gen_grad_clip: 1.0
disc_grad_clip: 1.0
# scheduler
gen_schedule_params:
warmup_step: 3000
down_step: 400000
min_lr: 2.0e-5 # this should be the final lr
max_lr: 1.0e-4
disc_schedule_params:
warmup_step: 3000
down_step: 400_000
min_lr: 2.0e-5 # this should be the final lr
max_lr: 1.0e-4
|