lmflow.pipeline.finetuner#
The Finetuner class simplifies the process of running finetuning process on a language model for a TunableModel instance with given dataset.
Attributes#
Classes#
Initializes the Finetuner class with given arguments. |
Module Contents#
- class lmflow.pipeline.finetuner.Finetuner(model_args, data_args, finetuner_args, *args, **kwargs)[source]#
Bases:
lmflow.pipeline.base_tuner.BaseTuner
Initializes the Finetuner class with given arguments.
- Parameters:
- model_argsModelArguments object.
Contains the arguments required to load the model.
- data_argsDatasetArguments object.
Contains the arguments required to load the dataset.
- finetuner_argsFinetunerArguments object.
Contains the arguments required to perform finetuning.
- argsOptional.
Positional arguments.
- kwargsOptional.
Keyword arguments.