Revert "Train"

This reverts commit a56c22488cf40ee3f919c56fec7e8a26f52e2f3d.