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.
baiyubaiyu / pytorch-utils Goto Github PK
View Code? Open in Web Editor NEWThis project forked from alejandrodebus/pytorch-utils
Useful functions to work with PyTorch. At the moment, there is a function to work with cross validation and kernels visualization.