Module audiocraft.solvers.audiogen

Classes

class AudioGenSolver (cfg: omegaconf.dictconfig.DictConfig)
Expand source code 
class AudioGenSolver(musicgen.MusicGenSolver):
    """Solver for AudioGen re-implementation training task.

    Note that this implementation does not strictly follows
    the method proposed in https://arxiv.org/abs/2209.15352
    but is derived from MusicGen's training pipeline.

    More information can be found in the AudioGen model card.
    """
    DATASET_TYPE: builders.DatasetType = builders.DatasetType.SOUND

Solver for AudioGen re-implementation training task.

Note that this implementation does not strictly follows the method proposed in https://arxiv.org/abs/2209.15352 but is derived from MusicGen's training pipeline.

More information can be found in the AudioGen model card.

Ancestors

Class variables

var DATASET_TYPE : DatasetType

Inherited members