Gradient spike when overfitting

<img width="2557" height="1268" alt="Image" src="https://github.com/user-attachments/assets/39aaa446-78d0-475c-9fea-18a107a37377" />

# Problem description
gradient norm becomes spiky when the model overfits.

# How to reproduce
1. open comfyResearch: https://online.comfy-research.com/
2. Type in 1910987607 in the input field
3. draw the random graph and run