Move value to device before logging for metric (#3649)

move value to device before logging for metric

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."""