use generator count for batch index as well, pretty neat