Spaces:
Runtime error
Runtime error
File size: 566 Bytes
fa6856c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
.. _trainers:
RL Trainers
*******************
RL Trainers are what you're training with trlX. Currently, we support PPO and ILQL.
Note that new trainers must be registered with ``trlx.trainer.register_trainer``.
**General**
.. autoclass:: trlx.trainer.BaseRLTrainer
:members:
.. autoclass:: trlx.trainer.accelerate_base_trainer.AccelerateRLTrainer
:members:
**PPO**
.. autoclass:: trlx.trainer.accelerate_ppo_trainer.AcceleratePPOTrainer
:members:
**ILQL**
.. autoclass:: trlx.trainer.accelerate_ilql_trainer.AccelerateILQLTrainer
:members:
|