:bug: Reduce the memory usage of the segmentation metrics

ENSTA-U2IS-AI / torch-uncertainty

Open-source framework for uncertainty and deep learning models in PyTorch :seedling:

https://torch-uncertainty.github.io

Apache License 2.0

308 stars 20 forks source link

:bug: Reduce the memory usage of the segmentation metrics #91

Closed o-laurent closed 6 months ago

o-laurent commented 7 months ago

Currently, validation may fail even for relatively small models such as DeepLabV3 when using large image crops. Calibration Error is the worst metric but others seem to also have huge impacts on the VRAM.