Open ZDDDYuan opened 3 months ago
Far from being an expert, but you dont have a training loop.
You just trained for "one" epoch on random input. See: text = torch.randint(0, 49408, (4, 256)).cuda() images = torch.randn(4, 3, 256, 256).cuda()
There is nothing to output other then noise
I'm using the code in the Usage section of the README, which produces a noise-filled image. Has anyone encountered a similar problem and how should I fix it?