Have a look at Skorch. It’s a scikit-learn compatible wrapper for PyTorch.
scikit itself offers a lot of cross-validation methods.
16 Likes
Have a look at Skorch. It’s a scikit-learn compatible wrapper for PyTorch.
scikit itself offers a lot of cross-validation methods.