andreys42:

Does Litghtning do the same for “on_train_epoch_end” callback or I have to set model.eval() manually?


I don’t know, you could find out yourself with assert not model.training (for eval mode) or so.


Which of pre-defined callbacks provided with model.eval “under the hood”?


The rule of thumb would be that training is set up with train and validation/testing with eval, but if you want this more detailed, I would probably check with in the lightning discussions 3.



 andreys42:

is it correct that training loss calculating using model in training mode?


I think the training loss is just computed during trianing, yes.