Fix bug in the eval function of the prior

lucidrains / DALLE2-pytorch

Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch

MIT License

11.03k stars 1.07k forks source link

Closed rom1504 closed 2 years ago

rom1504 commented 2 years ago

https://github.com/lucidrains/DALLE2-pytorch/blob/main/train_diffusion_prior.py#L17 this function should compute a single averaged number and report it for the current training step the current behavior is very confusing and wrong

lucidrains commented 2 years ago

@rom1504 oh oops, missed this in the review, should be good now, i think

rom1504 commented 2 years ago

thanks