Module audiocraft.solvers
Solvers. A Solver is a training recipe, combining the dataloaders, models, optimizer, losses etc into a single convenient object.
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.
"""
Solvers. A Solver is a training recipe, combining the dataloaders, models,
optimizer, losses etc into a single convenient object.
"""
# flake8: noqa
from .audiogen import AudioGenSolver
from .builders import get_solver
from .base import StandardSolver
from .compression import CompressionSolver
from .musicgen import MusicGenSolver
from .diffusion import DiffusionSolver
Sub-modules
audiocraft.solvers.audiogen
audiocraft.solvers.base
audiocraft.solvers.builders
-
All the functions to build the relevant solvers and used objects from the Hydra config.
audiocraft.solvers.compression
audiocraft.solvers.diffusion
audiocraft.solvers.magnet
audiocraft.solvers.musicgen