多gpu训练问题 - Githubissues

begeekmyfriend / tacotron2

Forked from NVIDIA/tacotron2 and merged with Rayhane-mamah/Tacotron-2

BSD 3-Clause "New" or "Revised" License

81 stars 38 forks source link

多gpu训练问题 #34

Open yannier912 opened 3 years ago

yannier912 commented 3 years ago

您好，请问下此版本可以用多gpu训练吗？我修改了hparams.py中num_gpus = 1，train.py中--distributed-run True，然后在train_tacotron2.sh中CUDA_VISIBLE_DEVICES=0,1。可是始终只用到了0号gpu，1上面没有。是还需要设置什么吗？

leijue222 commented 3 years ago

此版本暂不支持多GPU训练，如需多GPU训练，请参考NVIDIA/tacotron2的相关代码自行修改。我在单1080Ti上也能训练出好的效果，且无需训练很长时间。几个小时内训练到20个epoch的时候log曲线就挺不错了，然后可以继续训练至200或者更多epoch以达到更好的除噪效果。