Are the Dataloader and Dataset classes written in pure python or do they use any c++ bindings?
They should use pure Python: dataset.py, dataloader.py.
dataset.py
dataloader.py