Package flashy
Expand source code
# flake8: noqa
__version__ = '0.0.1a1'
from . import distrib, adversarial
from .formatter import Formatter
from .logging import ResultLogger, LogProgressBar, bold, setup_logging
from .solver import BaseSolver
from .utils import averager
Sub-modules
flashy.adversarial
-
For training adversarial losses, we provide an AdversarialLoss wrapper that encapsulate the training of the adversarial loss. This allows us to keep …
flashy.distrib
flashy.formatter
-
Formatter takes care of formatting metrics for display in the logs. For each possible training stage, it takes a mapping from metric pattern to …
flashy.loggers
flashy.logging
flashy.solver
-
Base Solver class. Specific solver should inherit this class. Solver takes care of various things, like setting up logging? As well as running stages.
flashy.state
-
Utility class for automatically handling state of solver. The object
StateManager()
can track stateful components of the solver. Each component … flashy.utils
-
Various utilities.