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