Hi @ptrblck,
I have a custom dataset that returns the image and a label with it.
I am getting the same error,
but its because I am returning the labels as well in the getitem method.
How do you suggest I handle this?
Thank You