refactor: extract training logic into dedicated training loop module 1f92b89 ydmhmhm commited on May 29