# augment inputs with noise
inputs = inputs + torch.randn_like(inputs, device='cuda') * noise_sd
Line 109 in train.py.
If there is no torch.clamp() after line 109, it is possible that the input image will exceed its allowed pixel value range (i.e., [0,1])