search

sanyam5 / arc-pytorch

The first public PyTorch implementation of Attentive Recurrent Comparators
MIT License
148 stars 30 forks source link

RGB image as input? #7

Open jmkim0309 opened 6 years ago

jmkim0309 commented 6 years ago

Hi, thanks for your great implementation on ARC. Is the current version also support 3-channel RGB images as input?

I found that the implementation works fine with gray-scale input images, such as Omniglot dataset, but when I tried with different dataset with RGB images, it seems like it doesn't work.

sanyam5 commented 6 years ago

Unfortunately, I have only implemented the most basic version. But I am open to pull requests.