greydanus / mnist1d

A 1D analogue of the MNIST dataset for measuring spatial biases and answering Science of Deep Learning questions.
Apache License 2.0
199 stars 33 forks source link

Pytorch DataSet for MNIST1D #12

Open psteinb opened 2 months ago

psteinb commented 2 months ago

Hi,

for a course, I wrote a pytorch compatible Dataset class for MNIST1D. The examples in this repo don't require one as such as most do not rely on a torch.utils.data.DataLoader. I am still wondering, if it may makes sense to provide a Dataset compliant class for easier handling of MNIST1D in pytorch based code.

I am happy to send a PR if need be, but wanted to discuss the matter first.

Best, Peter

greydanus commented 2 months ago

A PR featuring this addition would be much appreciated :)