跳转至

Utilities

danling.runner.utils

RunnerMode

Bases: LowercaseStrEnum

RunnerMode is an enumeration of running modes.

Attributes:

Name Type Description
train

Training mode.
eval

Evaluation mode.
inf

Inference mode.
Source code in danling/runner/utils.py
Python
49
50
51
52
53
54
55
56
57
58
59
60
61
class RunnerMode(StrEnum):  # pylint: disable=too-few-public-methods
    r"""
    `RunnerMode` is an enumeration of running modes.

    Attributes:
        train: Training mode.
        eval: Evaluation mode.
        inf: Inference mode.
    """

    train = auto()
    eval = auto()
    inf = auto()

on_main_process

Python
1
on_main_process(func)

Decorator to run func only on main process.

Source code in danling/runner/utils.py
Python
102
103
104
105
106
107
108
109
110
111
112
113
def on_main_process(func):
    """
    Decorator to run func only on main process.
    """

    @wraps(func)
    def wrapper(self, *args, **kwargs) -> Any | None:
        if self.is_main_process or not self.distributed:
            return func(self, *args, **kwargs)
        return None

    return wrapper

on_local_main_process

Python
1
on_local_main_process(func)

Decorator to run func only on local main process.

Source code in danling/runner/utils.py
Python
116
117
118
119
120
121
122
123
124
125
126
127
def on_local_main_process(func):
    """
    Decorator to run func only on local main process.
    """

    @wraps(func)
    def wrapper(self, *args, **kwargs) -> Any | None:
        if self.is_local_main_process or not self.distributed:
            return func(self, *args, **kwargs)
        return None

    return wrapper