lucidrains / nuwa-pytorch

Implementation of NÜWA, state of the art attention network for text to video synthesis, in Pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Type of dataset for training VQ-GAN

antonibigata opened this issue · comments

Hi,

First, thanks a lot for the amazing work!
I have one question regarding the training of the VQ-GAN, do you recommend training it on a dataset similar to the dataset the nuwa model will be trained?
What I mean is, if I want to train nuwa to generate sport videos based on text, do I need to also train the VQ-GAN on a sport dataset?

Thanks a lot