Fix EfficientNet AMP NaN grads (FP16-only guard)#170

Open

nkundiushuti wants to merge 1 commit intomainfrom

marius/nan-bug-retraining

Commits on Mar 27, 2026

amp guard in efficientnet to avoid nan gradients in the case of fp16
nkundiushuti
committed