(self, writer, global_step, prefix="")
| 430 | return self.V / self.N |
| 431 | |
| 432 | def write(self, writer, global_step, prefix=""): |
| 433 | writer.add_scalar(os.path.join(prefix, "PSNR"), self.measure(), global_step) |
| 434 | |
| 435 | def report(self): |
| 436 | return f'PSNR = {self.measure():.6f}' |