Problem about predict nan after few batch

2

I use tensorflow.keras reslove this problem, so maybe change tenorflow is better.