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, | |