Make metrics device configurable through `METRICS_DEVICE` env variable

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

Make metrics device configurable through `METRICS_DEVICE` env variable #683

Closed nkaenzig closed 1 month ago

nkaenzig commented 1 month ago

Closes #677

Segmentation metrics calculation runs much faster on GPU - tested on mps and A100 gpu - Therefore the new default is to use the same device as for training for the metrics calculation.