bridge.training.finetune#

Module Contents#

Functions#

finetune

Main function to run the finetuning.

API#

bridge.training.finetune.finetune(
config: megatron.bridge.training.config.ConfigContainer,
forward_step_func: Callable,
) → None#

Main function to run the finetuning.

Parameters:

  • config – The main configuration container holding all necessary parameters.

  • forward_step_func – A callable that performs a single forward and backward step, returning the loss and any computed metrics.

.. warning::

This is an experimental API and is subject to change in backwards incompatible ways without notice.