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

print metrics report

This commit is contained in:
shahules786 2022-10-11 21:30:13 +05:30
parent abcdc29309
commit a57a7ff0fc
1 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
enhancer/models/model.py
View File

@ -230,13 +230,10 @@ class Model(pl.LightningModule):
for metric in test_mean_metrics.keys(): for metric in test_mean_metrics.keys():
test_mean_metrics[metric] /= len(outputs) test_mean_metrics[metric] /= len(outputs)
for k, v in test_mean_metrics.items(): print("----------TEST REPORT----------\n")
self.logger.experiment.log_metric( for metric in test_mean_metrics.keys():
run_id=self.logger.run_id, print(f"|{metric.upper()} | {test_mean_metrics[metric]} |")
key=k, print("--------------------------------")
value=v,
step=self.current_epoch,
)
def on_save_checkpoint(self, checkpoint): def on_save_checkpoint(self, checkpoint):