PyMarlin/pymarlin/utils/writer/base.py at 4b6f2ba7b8ec5849c67d26c5de4726fa6e056e4a · microsoft/PyMarlin · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
"""
Base class for Writers
"""
from abc import ABC
import dataclasses

@dataclasses.dataclass
class WriterInitArguments:
    """
    Writer Arguments.
    Args:
        tb_log_dir(str): Defaults to 'logs'
        tb_logpath_parent_env(str): Defaults to None
        tb_log_multi(bool): Defaults to False
        tb_log_hist_steps(int): Defaults to 20000
        model_log_level(str): Defaults to 'INFO'
    """
    tb_log_dir: str = 'logs'
    tb_logpath_parent_env: str = None
    tb_log_multi: bool = False
    tb_log_hist_steps: int = 20000
    model_log_level: str = 'INFO'

class Writer(ABC):
    """
    Abstract Base class for Writers.
    """
    def __init__(self, logger):
        self.logger = logger

    def log_scalar(self, k, v, step):
        pass

    def log_multi(self, k, v, step):
        pass

    def log_model(self, flat_weights, flat_grads, step):
        pass

    def log_args(self, args):
        pass

    def log_graph(self, model, device=None):
        pass

    def log_image(self, k, v, step, dataformats='HW'):
        pass

    def log_pr_curve(self, k, preds, labels, step):
        pass

    def log_histogram(self, param_name, vals, step):
        pass

    def log_embedding(self, tag, mat, labels, step):
        pass

    def _log_norms(self, step, param_name, weight_norm, grad_norm):
        pass

    def log_multi_line(self, string):
        lines = string.split('\n')
        for line in lines:
            self.logger.info(line)

    def flush(self):
        pass

    def finish(self):
        pass