Christian_Ses:

In training mode I use a batch size of 12 and in evaluation mode I use a batch size of 1. Is this a problem? But im always calculating the mean of the whole validation epoch for plotting the epochs.


No it shouldn’t be a problem as long as you do model.train() and model.eval()