ejhung

I'm also working on reproducing the training step according to the author's manual. I tried to revise the issue of train.py last week, so not sure my fix would resolve...