search

CSAILVision / gandissect

Pytorch-based tools for visualizing and understanding the neurons of a GAN. https://gandissect.csail.mit.edu/
MIT License
1.77k stars 283 forks source link

Custom dataset #12

Open FrederikWarburg opened 5 years ago

FrederikWarburg commented 5 years ago

Hi

I'm trying to train on my own dataset. I created a data loader class like this

Screenshot 2019-03-26 at 21 33 04

I call the dissect model like this

Screenshot 2019-03-26 at 21 35 40

But I keep getting an indexing error.

Screenshot 2019-03-26 at 21 36 12

Can anybody help me with this issue?