Convert dataset to pytorch dataloader

[lhoestq]

@pfjf2

Also, for whatever reason, it’s not caching the .map() transform, even though the hash for the lambda is the same on every session. What’s going on there?

Caching only happens with datastes on disk. On the other hand, nothing is written on disk or cached for datasets in streaming mode