fix: patch torchmetrics._apply using CPU probe tensor instead of self.device ec4688a will702 commited on Mar 23
fix: use callable map_location to skip Lightning model.to() — avoids torchmetrics CUDA assertion 49e908a will702 commited on Mar 23
fix: patch torchmetrics._apply for CPU-only load of GPU-trained checkpoint c2a87a5 will702 commited on Mar 23