nat.finetuning.finetuning_runtime

Finetuning runtime for NAT that orchestrates the training process.

Attributes

logger

Functions

run_finetuning(→ None)	Run finetuning based on the provided configuration.
finetuning_main(→ None)	Main entry point for finetuning runtime.
run_finetuning_sync(→ None)	Synchronous wrapper for running finetuning.

Module Contents

logger

async run_finetuning(

runner: nat.finetuning.interfaces.finetuning_runner.Trainer,

) → None

Run finetuning based on the provided configuration.

Args:

runner: An instance of the Trainer to run finetuning with

async finetuning_main(

run_config: nat.data_models.finetuning.FinetuneRunConfig,

) → None

Main entry point for finetuning runtime.

Args:

run_config: FinetuneRunConfig object containing finetuning settings

run_finetuning_sync(

run_config: nat.data_models.finetuning.FinetuneRunConfig,

) → None

Synchronous wrapper for running finetuning.

Args:

run_config: FinetuneRunConfig object containing finetuning settings