跳转至

Utilities

danling.runner.utils

RunnerMode

Bases: LowercaseStrEnum

RunnerMode is an enumeration of running modes.

Attributes:

Name Type Description
train

Training mode.
evaluate

Evaluation mode.
infer

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

    Attributes:
        train: Training mode.
        evaluate: Evaluation mode.
        infer: Inference mode.
    """

    train = auto()
    evaluate = auto()
    infer = auto()

on_main_process

Python
on_main_process(func)

Decorator to run func only on main process.

Source code in danling/runner/utils.py
Python
105
106
107
108
109
110
111
112
113
114
115
116
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
on_local_main_process(func)

Decorator to run func only on local main process.

Source code in danling/runner/utils.py
Python
119
120
121
122
123
124
125
126
127
128
129
130
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