before_forward callback confusing naming

In the eval loop, before_eval_forward is called, but in the training loop, before_forward is called. It can be confusing since it's not clear whether this callback is used in the train loop or eval loop from a user point of view. Maybe we should stick to naming before_train_forward and after_train_forward just like it's done in the evaluation loop.