trlx/docs/source/pipeline.rst

.. _pipeline:

Pipelines
************************

Pipelines are how you read from a dataset with trlX. Rollout stores are how models store experiences created
for them. It is these experiences in their rollout store that they are trained on.

**General**

.. autoclass:: trlx.pipeline.BasePipeline
    :members:

.. autoclass:: trlx.pipeline.BaseRolloutStore
    :members:

**PPO**

.. autoclass:: trlx.pipeline.ppo_pipeline.PPORolloutStorage
    :members:

**ILQL**

.. autoclass:: trlx.pipeline.offline_pipeline.PromptPipeline
    :members:

.. autoclass:: trlx.pipeline.offline_pipeline.ILQLRolloutStorage
    :members: