add agent trainer to switch between trial and steps training count