Implement averaging over final epoch