log metric

enhancer/models/model.py

@@ -181,9 +181,11 @@ class Model(pl.LightningModule):
             value = metric(target, prediction)
             metric_dict[metric.name] = value
 
-        self.logger.experiment.log_metrics(
+        for k, v in metric_dict.items():
+            self.logger.experiment.log_metric(
                 run_id=self.logger.run_id,
-            metrics=metric_dict,
+                key=k,
+                value=v,
                 step=self.global_step,
             )