In this repository there are useful functions to work with Pytorch. The functions are made to work with medical images, but are easily adaptable to other types of problems.
alejandrodebus/Pytorch-Utils
Useful functions to work with PyTorch. At the moment, there is a function to work with cross validation and kernels visualization.
Python