File size: 402 Bytes
60b9091 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
lightning:
callbacks:
image_logger:
target: train.ImageLogger
params:
for_specs: true
vocoder_cfg:
target: train.VocoderMelGan
params:
ckpt_vocoder: /apdcephfs_cq2/share_1297902/speech_user/shaunxliu/dongchao/code/VQ-VAE/vocoder/logs/vggsound/
trainer:
sync_batchnorm: true
distributed_backend: ddp
gpus: 0,1,2,3,4,5,6,7,
|