mirror of
https://github.com/PaddlePaddle/PaddleOCR.git
synced 2025-08-09 00:58:23 +00:00
18 lines
554 B
Python
18 lines
554 B
Python
|
from .base_logger import BaseLogger
|
||
|
|
from visualdl import LogWriter
|
||
|
|
|
||
|
|
class VDLLogger(BaseLogger):
|
||
|
|
def __init__(self, save_dir):
|
||
|
|
super().__init__(save_dir)
|
||
|
|
self.vdl_writer = LogWriter(logdir=save_dir)
|
||
|
|
|
||
|
|
def log_metrics(self, metrics, prefix=None, step=None):
|
||
|
|
if not prefix:
|
||
|
|
prefix = ""
|
||
|
|
updated_metrics = {prefix + "/" + k: v for k, v in metrics.items()}
|
||
|
|
|
||
|
|
for k, v in updated_metrics.items():
|
||
|
|
self.vdl_writer.add_scalar(k, v, step)
|
||
|
|
|
||
|
|
def close(self):
|
||
|
|
self.vdl_writer.close()
|