How to replace dataset and change dimensions

[shrybht]

Hi,
This implementation was one of the easiest to implement as well as understand. I am curious and would want to implement COCO dataset using the same code.

1. How do I change the dimensions to 3?
2. How to import COCO dataset into the code? I tried importing from torchvision.datasets but it is throwing an error constantly.
3. An overview as to how to make it work on COCO dataset.

Could u please help me with this?

[theaaron123]

Hi,

If you look at the svhn implementation I added and compare to the original this should give you an intuition on what needs changing for a new dataset