This post seems to give an additional example of how to use cross validation in skorch
.
Alternatively you could also directly use sklearn.model_selection
methods to create indices for the splits and recreate the datasets via Subset
s in each iteration.