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:
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: