It would be easier if Streaming (IterableDataset) could be used, but I don’t know how to enable Streaming in AutoTrainAdvanced…
IterableDataset
It might be quicker to split the dataset into several smaller datasets in advance.