recon images is black

[syyxsxx]

i train magvit2 on imagenet
the recon images is black, but the loss looks normal

[nangying1112]

I met the same problem. Have you solved this?

[dyflional10]

I met the same problem. Have you solved this?
I also encountered the same problem where the loss remains constant, and I discovered that the gradient is NaN.

[syyxsxx]

I turned off mixed precision training，solve this problem

[lucidrains]

@syyxsxx thank you for finding and sharing the source of the issue. i'll figure out what's the matter with mixed precision..

[dyflional10]

@lucidrains Thanks for your great work. I turned off mixed precision training, the loss looks normal, but the image reconstruction is abnormal.

[misery0424]

I turned off mixed precision training，solve this problem

can you share your config and training log, thanks

[VoyageWang]

I also met the same problem the construction image is not so normal. Can you share the details of solving this problem? How to turn off the mixed precision traning

I turned off mixed precision training，solve this problem

[J-Wu97]

我在 imagenet 上训练 magvit2， 侦察图像是黑色的，但损失看起来很正常

hello，Have you solved this problem?

[kxgong]

Hi, I have a question. Why use images to train magvit-2? Should it designed for videos?