My program always stops with no error, just being at a standstill, after working some epochs. And the same program works well in another dataset. By the way, GPU just run this  only program. Does anybody know why? Could you give me some advice? I really appreciate that.