Skip to content
GitHub
View on GitHub

TrainConfig

Compose dataset, model, and recipe into one training entrypoint.

from modal_training_gym.common.train import TrainConfig

Compose dataset, model, and recipe into one training entrypoint.

FieldTypeDefaultDescription
datasetDatasetConfig
modelModelConfig
recipemodal_training_gym.train_recipes.base.BaseTrainRecipe
checkpointmodal_training_gym.common.checkpoint.Checkpoint | NoneNone
detachboolTrue
_stable_idstr | NoneNone

recipe_param_summary(self) -> dict[str, dict[str, typing.Any]]

Section titled “recipe_param_summary(self) -> dict[str, dict[str, typing.Any]]”

train(self) -> modal_training_gym.common.train_result.TrainResult

Section titled “train(self) -> modal_training_gym.common.train_result.TrainResult”

Build the app, run training, and return the TrainResult.

Source: modal_training_gym/common/train.py