For example, according to the Chinchilla paper, given a Model size and FLOPs budget, this functionally should estimate the dataset sizes to test.