Two tiny changes to train/eval_lm. For train fix an off by one, while for...

Two tiny changes to train/eval_lm. For train fix an off by one, while for eval_lm make it work when the task is translation'