Hello,
First of all thank you for this package, it is a game changer! I was wondering if there is a preferred way of using grayscale image datasets, like MNIST, with FFCV. Since torchvision dataset yields PIL images, I don't think it is suitable to use BytesField or NDArrayField in the DatasetWriter. I was able to use ffcv with MNIST by manually adding a num_channels parameter to image decoders and commenting out this line in the RGBImageField but I was wondering if there was a built-in way that I am missing.
Thanks
Hello,
First of all thank you for this package, it is a game changer! I was wondering if there is a preferred way of using grayscale image datasets, like MNIST, with FFCV. Since torchvision dataset yields PIL images, I don't think it is suitable to use
BytesFieldorNDArrayFieldin theDatasetWriter. I was able to use ffcv with MNIST by manually adding anum_channelsparameter to image decoders and commenting out this line in theRGBImageFieldbut I was wondering if there was a built-in way that I am missing.Thanks