CSAILVision/gandissect

Custom dataset

FrederikWarburg opened this issue · 0 comments

FrederikWarburg commented

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?