karmus89:

IMHO you should maybe refrain from passing a DataLoader and instead let Skorch’s NeuralNet take care of wrapping the Dataset to a DataLoader . See their quick tutorial here.


thank for your reply I also got this answer and it works https://stackoverflow.com/questions/56490278/passing-tensordataset-or-dataloader-to-skorch/56667050#56667050 510