lmflow.pipeline.rm_tuner
========================

.. py:module:: lmflow.pipeline.rm_tuner


Attributes
----------

.. autoapisummary::

   lmflow.pipeline.rm_tuner.logger


Classes
-------

.. autoapisummary::

   lmflow.pipeline.rm_tuner.RewardModelTuner


Module Contents
---------------

.. py:data:: logger

.. py:class:: RewardModelTuner(model_args, data_args, finetuner_args, *args, **kwargs)

   Bases: :py:obj:`lmflow.pipeline.finetuner.Finetuner`


   Initializes the `RewardModelTuner` class.

   :param model_args: Contains the arguments required to load the model.
   :type model_args: ModelArguments object.
   :param data_args: Contains the arguments required to load the dataset.
   :type data_args: DatasetArguments object.
   :param finetuner_args: Contains the arguments required to perform finetuning.
   :type finetuner_args: RewardModelTunerArguments object.
   :param args: Positional arguments.
   :type args: Optional.
   :param kwargs: Keyword arguments.
   :type kwargs: Optional.


   .. py:method:: tune(model: lmflow.models.hf_text_regression_model.HFTextRegressionModel, dataset, transform_dataset_in_place=True, data_collator=None, **kwargs)