OpenRL - 统一的强化学习框架
OpenRL是一个开源的通用强化学习研究框架,支持单智能体、多智能体、离线RL、自对弈和自然语言等多种任务的训练。它基于PyTorch开发,旨在为强化学习研究社区提供一个简单易用、灵活高效且可持续的平台。
🚀 快速入门
- 安装OpenRL:
pip install openrl
- 简单示例:
from openrl.envs.common import make
from openrl.modules.common import PPONet as Net
from openrl.runners.common import PPOAgent as Agent
env = make("CartPole-v1", env_num=9)
net = Net(env)
agent = Agent(net)
agent.train(total_time_steps=20000)
只需几行代码,就可以完成强化学习训练!