Fix all bugs will LBFGSTrainer