Solved by ottonemo in post #2 
              
              
                As you may know, you can use datasets or even your own dataloader with skorch but this is a bit problematic in conjunction with GridSearchCV or other parameter searches since they expect indexable inputs (torch datasets aren’t). For this reason skorch has a SliceDataset. You can use it as follows: 
g…