Training loop break and terminate when using customize data

[suikei-wang]

Hi there, I tried to train with my own data using this code. Some samples work well. However, I found that for some data, this error occurs:

The training is done right after it started. During debug, I found that the training loop is break out in the first iteration. Any ideas to sort it out? Thank you!