EnlightenGAN icon indicating copy to clipboard operation
EnlightenGAN copied to clipboard

why loss is nan

Open Kelsey2018 opened this issue 2 years ago • 0 comments

Hi, I have input data between 0 and 1, and at the first several epoches, the loss is normal, but then it turned to nan. I cannot find the errors,can you help me? in base_dataset.py transform_list += [transforms.ToTensor(), # normalized to [0, 1] transforms.Normalize((0, 0, 0), (1, 1, 1))] in util.py line 18 image_numpy = np.transpose(image_numpy, (1, 2, 0)) * 255.0

Kelsey2018 avatar Apr 14 '22 11:04 Kelsey2018