TrainPOS1: Fix arguments to train method to make it compile with master.