From 600c0378afe1b7bb876f593e7439245860642e42 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Nicolas=20K=C3=A4nzig?= <36882833+nkaenzig@users.noreply.github.com> Date: Mon, 14 Oct 2024 16:38:23 +0200 Subject: [PATCH] set default metrics device to cpu (#688) --- src/eva/core/models/modules/module.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/eva/core/models/modules/module.py b/src/eva/core/models/modules/module.py index ba65f679..10eddeae 100644 --- a/src/eva/core/models/modules/module.py +++ b/src/eva/core/models/modules/module.py @@ -50,7 +50,7 @@ def default_postprocess(self) -> batch_postprocess.BatchPostProcess: @property def metrics_device(self) -> torch.device: """Returns the device by which the metrics should be calculated.""" - device = os.getenv("METRICS_DEVICE", None) + device = os.getenv("METRICS_DEVICE", "cpu") return self.device if device is None else torch.device(device) @override