Running a session to train