关于在训练过程中在不同样本之间管理状态的问题

你好! 感谢您分享这个优秀的实现。 该架构非常优雅，我从您的代码中学到了很多东西。 我有一个关于训练过程中携带状态管理的问题，我想讨论一下。