Package `flashy`

Expand source code

# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.

# flake8: noqa

__version__ = '0.0.3a'

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: Distrib utilities. When running from inside Dora, you can use the init function imported from dora.distrib. Also works outside of Dora if …
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: Logging related utilities.
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.