multiple epochs iterator improvements