关于多机训练
shanhaidexiamo opened this issue · comments
您好。关于vqgan等模型的训练，我运行了python cli.py fit -c configs/config_gan.yaml ，并且把cli.py里改成了MegaGANTrainer，单机可运行。此外，我注意到您的config里有 strategy: ddp等字样，是用于配置lightning多机训练的参数。但是开启后会报错：
意思是每个gpu的模型参数存在不一致，请问该问题您是否遇到过？
VQ does not support DDP, you can replace it with your vector quantification or wait for a new commit.
Fixed on branch sr24k.