Fixed a bug in calculating mse with limited instances per epoch.