Allow IterableDataset to be passed to LightningDataModule.from datasets
Allow IterableDataset to be passed to LightningDataModule.from datasets: a task in MiMo-V2.6-RL-oss: Agentic RL Environments (MiMo RL release). 🚀 Feature Allow IterableDataset to be passed to LightningDataModule.from datasets Motivation Currently, we cannot pass IterableDataset instances to…
The task
🚀 Feature Allow IterableDataset to be passed to LightningDataModule.from_datasets Motivation Currently, we cannot pass IterableDataset instances to LightningDataModule's from_datasets method. This is because because IterableDataset cannot be used with DataLoader instances that…
Part of XiaomiMiMo/MiMo-V2.6-RL-oss.