RLPack provides in-built models that can be used on the fly with simply keywords from config. This help you set up experiments and train your agent on the fly.
To include in-built models, you need to pass the key model_name
in the config. An example to use
Currently, the following models are in-built into RLPack:
In-Built model | Keyword |
---|---|
MLP | "mlp" |
A2C MLP | "actor_critic_mlp_policy" |