TorchRL学习
TorchRL
TorchRL 是一个用于 PyTorch 的开源强化学习 (RL) 库。TorchRL 提供了 PyTorch 和 Python 优先的、低级别和高级别的 RL 抽象,旨在实现高效、模块化、有文档和经过适当测试。其代码旨在支持 RL 研究。大部分代码使用高度模块化的方式用 Python 编写,以便研究人员可以轻松地替换组件、对其进行转换或轻松编写新组件。
入门
Env, TED, Transform
创建环境
创建环境通常与底层后端 API 一样简单。以下是使用 gym 的示例。
from torchrl.envs import GymEnv
env = GymEnv("Pendulum-v1")
运行环境
评论