RElbers / region-mutual-information-pytorch

PyTorch implementation of the Region Mutual Information Loss for Semantic Segmentation.
MIT License
25 stars 1 forks source link

torch.cholesky replaced with torch.linalg.cholesky #5

Closed daandres closed 1 year ago

daandres commented 1 year ago

torch library shows a warning in 2.1 to the usage of torch.cholesky. This is fixed by changing to the new call of torch.linalg.cholesky