Error during training

[zi1zi]

Hello, I used the mx-font training data set in this project. There are 36 fonts in total, including 13 XIAOZHUANTI, and 800 pictures for each font. However, during the 14-15w iter period, the following error was reported: ValueError: matrix contains Invalid numeric entries. After running several times, the error occurred during 14-15w iter.It feels like there is a mode collapse，But don't know the reason

On this basis, I improved some methods, including adding some attention, but when it came to 4w iter, the above error occurred.

[zi1zi]