search

kaiko-ai / eva

Evaluation framework for oncology foundation models (FMs)
https://kaiko-ai.github.io/eva/
Apache License 2.0
66 stars 4 forks source link

Update `metrics_device` to use `cpu` when running on `mps` #691

Closed nkaenzig closed 1 month ago

nkaenzig commented 1 month ago

We found compatibility issues, particularly when calculating the segmentation metrics (e.g. GeneralizedDice) on mps (yields invalid / out of range or even negative numbers). This issue doesn't occur when running on cuda or cpu.