中文

Tianshou:一个高度模块化的深度强化学习库

机器学习 2022-08-11 v3

摘要

本文介绍 Tianshou,一个以 PyTorch 为后端、高度模块化的深度强化学习(DRL)Python 库。Tianshou 旨在通过提供灵活可靠的 DRL 算法基础设施而利于研究。它通过统一接口支持在线与离线训练,涵盖 20 余种经典算法。为推进相关研究并证明 Tianshou 的可靠性,我们发布了 Tianshou 在 MuJoCo 环境下的基准测试,覆盖八种具最优性能的经典算法。我们在 https://github.com/thu-ml/tianshou/ 开源了 Tianshou。

关键词

引用

@article{arxiv.2107.14171,
  title  = {Tianshou: a Highly Modularized Deep Reinforcement Learning Library},
  author = {Jiayi Weng and Huayu Chen and Dong Yan and Kaichao You and Alexis Duburcq and Minghao Zhang and Yi Su and Hang Su and Jun Zhu},
  journal= {arXiv preprint arXiv:2107.14171},
  year   = {2022}
}