@ptrblck please can you tell me why my model outputting nan?
if i use same model without groupnormalization code then i don’t get any  nan, only thing i changed in my code is bn to gn and nothing else! why my model is outputting nan after using groupnormalization?