Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

log metric

This commit is contained in:
shahules786 2022-10-10 15:32:37 +05:30
parent a417e226f3
commit 81c5f13ff6
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
enhancer/models/model.py
View File

@ -181,9 +181,11 @@ class Model(pl.LightningModule):
value = metric(target, prediction) value = metric(target, prediction)
metric_dict[metric.name] = value 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, run_id=self.logger.run_id,
metrics=metric_dict, key=k,
value=v,
step=self.global_step, step=self.global_step,
) )