Created by: jacquesqiao

First, instead of declaring a Trainer and call trainer.train(), is there any way to call the train_op only once (aka stepping)