Training problem

[baiguiang]

Thank you for wonderful work.

I am training a model on my dataset, I can train the first epoch normally, but it will get stuck after the first epoch is finished. It seems that the value of batch_size is also strange, can you explain?

Kindly help me on this.

Thanks