loss Nan

[wylblank]

我在MUNIT上用自己的数据集进行训练，因为显存不足（我的显卡是2070，8G），所以将crop_image_height和crop_image_wigth修改为256*256，在这种情况下如果vgg_w不等于0，会loss就会出现Nan，请问应该怎么解决？非常感谢！！！

I used my own data set for training on MUNIT, because of insufficient video memory (my video card is 2070, 8G), so I changed the crop_image_height and crop_image_wigth to 256*256, in this case, if vgg_w is not equal to 0, loss will appear Nan. What should I do? Thank you so much!!