diff --git a/src/otx/core/model/base.py b/src/otx/core/model/base.py index 5042747e87b..89862020baa 100644 --- a/src/otx/core/model/base.py +++ b/src/otx/core/model/base.py @@ -349,7 +349,7 @@ def _log_metrics(self, meter: Metric, key: Literal["val", "test"], **compute_kwa warnings.warn(msg, stacklevel=1) continue - self.log(log_metric_name, value, sync_dist=True, prog_bar=True) + self.log(log_metric_name, value.to(self.device), sync_dist=True, prog_bar=True) def on_save_checkpoint(self, checkpoint: dict[str, Any]) -> None: """Callback on saving checkpoint."""