Skip to content

Add torch dataloader to hydra_models

While datasets usually have to be written manually by deriving from the torch Dataset class, the DataLoader usually is used just as a drop-in to get torchs functionality of batching, prefatching etc. This MR adds a DataLoader which can be initialized by hydra for these exact use-cases.

Closes #27 (closed)

Merge request reports