Yes. And because we are producing one Problem object at a time with IterableDataset, is there any benefit of making this object into IterableDataset ? If I may add to the question, how will the object be scattered if we are using DistributedDataParallel?