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():
test_mean_metrics[metric] /= len(outputs)
for k, v in test_mean_metrics.items():
self.logger.experiment.log_metric(
run_id=self.logger.run_id,
key=k,
value=v,
step=self.current_epoch,
)
print("----------TEST REPORT----------\n")
for metric in test_mean_metrics.keys():
print(f"|{metric.upper()} | {test_mean_metrics[metric]} |")
print("--------------------------------")
def on_save_checkpoint(self, checkpoint):