Spaces:
Runtime error
Runtime error
File size: 782 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 27 28 29 30 31 32 33 34 35 |
.. _configs:
Configs
************************
Training a model in TRL will require you to set several configs:
ModelConfig, which contains general info on the model being trained. TrainConfig, which contains things like
training hyperparameters. And finally, MethodConfig, which contains hyperparameters or settings for
the specific method being used (i.e. ILQL or PPO)
**General**
.. autoclass:: trlx.data.configs.TRLConfig
:members:
.. autoclass:: trlx.data.configs.ModelConfig
:members:
.. autoclass:: trlx.data.configs.TrainConfig
:members:
.. autoclass:: trlx.data.method_configs.MethodConfig
:members:
**PPO**
.. autoclass:: trlx.data.method_configs.PPOConfig
:members:
**ILQL**
.. autoclass:: trlx.data.method_configs.ILQLConfig
:members:
|