Hello everyone,
I am quite new with Skorch, and I am in the process of applying K-fold cross-validation using data Loaders from a directory of images.
After creating the train and validation loaders, I user the fit(X,y) method from NeuralNet class, with X = current_train_loader, and y = current_train_target. As soon as the fit(X,y) executes, the following error occurs:

image.png987×278 12.7 KB

Whereas, I fill in the input and target exactly as mentioned. Is there anything wrong with my procedure? Following is what my training phase looks like: