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.