pytorch

1
import logging
2
import sys
3
from types import ModuleType
4
from typing import Set
5

6
# log level (levels print what it says + all levels listed below it)
7
# DEBUG print full traces <-- lowest level + print tracing of every instruction
8
# INFO print compiler functions + distributed graphs
9
# WARN print warnings
10
# ERROR print exceptions
11
log_level: int = logging.DEBUG
12
# Verbose will print full stack traces on warnings and errors
13
verbose = False
14

15
# the name of a file to write the logs to
16
log_file_name: None = None
17

18

19
class _AccessLimitingConfig(ModuleType):
20
    def __setattr__(self, name, value) -> None:
21
        if name not in _allowed_config_names:
22
            raise AttributeError(f"{__name__}.{name} does not exist")
23
        return object.__setattr__(self, name, value)
24

25

26
_allowed_config_names: Set[str] = {*globals().keys()}
27
sys.modules[__name__].__class__ = _AccessLimitingConfig
28