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}
}