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 averagerSub-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.distribflashy.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.loggersflashy.loggingflashy.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.