ReferenceFull Library ReferenceNemo AutomodelNemo AutomodelRecipesMultimodalnemo_automodel.recipes.multimodal.pretrain

nemo_automodel.recipes.multimodal.pretrain

View as Markdown

Module Contents

Classes

NameDescription
PretrainRecipeForMultimodalPretraining entrypoint for multimodal BAGEL training.

API

class nemo_automodel.recipes.multimodal.pretrain.PretrainRecipeForMultimodal()

Bases: FinetuneRecipeForMultimodal

Pretraining entrypoint for multimodal BAGEL training.

The shared implementation lives in FinetuneRecipeForMultimodal because the BAGEL training step is the same; pretraining behavior is selected by the YAML model initializer and data mixture.