Skip to content

Commit

Permalink
set default metrics device to cpu (#688)
Browse files Browse the repository at this point in the history
  • Loading branch information
nkaenzig authored Oct 14, 2024
1 parent 209b694 commit 600c037
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/eva/core/models/modules/module.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 600c037

Please sign in to comment.