Cross-validation in pytorch

How to beautifully implement cross-validation in pytorch, while using different transformers for validation and training selection?
Are there solutions that only change the dataloader iterator, that is, do not require many changes in the fitting code?

skorch is a sikit-learn compatible PyTorch wrapper, which should allow you to apply the scikit-learn cross-validation utilities directly.

1 Like