Tianshou(天授) 基于 PyTorch 的强化学习平台开源项目

我要开发同款
匿名用户2020年03月31日
206阅读

技术信息

开源地址
https://github.com/thu-ml/tianshou
授权协议
MIT

作品详情

Tiashou(天授)是纯基于 PyTorch的强化学习平台,与现有的主要基于TesorFlow的强化学习库不同,Tiashou没有繁杂的嵌套类、不友好的API和速度较慢的代码,其提供了用于构建深度强化学习代理的快速框架和pythoicAPI。

支持的接口算法包括:

PolicyGradiet(PG)

DeepQ-Network(DQN)

DoubleDQN(DDQN) with-stepreturs

AdvatageActor-Critic(A2C)

DeepDetermiisticPolicyGradiet(DDPG)

ProximalPolicyOptimizatio(PPO)

TwiDelayedDDPG(TD3)

SoftActor-Critic(SAC)

Tiashou还支持所有算法并行,所有算法都重新格式化为基于重放缓冲(replay-buffer)模式。

特性:

快速:使用 laptop(i7-8750H+GTX1060)测试为例,CartPole-v0任务中,Tiashou仅需要3秒钟即可根据vailla策略梯度来训练代理:pytho3test/discrete/test_pg.py--seed0--reder0.03

            

          性能对比:

      

可重复性:具有单元测试,并且单元测试覆盖针对所有已实现算法的完整代理培训过程。单元测试确保了平台的可重复性模块化:所有算法分解为4部分:__iit__:初始化process_f:从重放缓冲区预处理数据(所有算法都重构为基于重放缓冲的算法)__call__:根据给定的观察值计算操作lear:从给定的批处理数据中学习优雅灵活:整体代码不到1500行,大多数已实现的算法少于100行Pytho代码。可以根据需要提供许多灵活的API。

功能介绍

Tianshou(天授)是纯基于 PyTorch 的强化学习平台,与现有的主要基于 TensorFlow 的强化学习库不同,Tianshou 没有繁杂的嵌套类、不友好的 API 和速度较慢的代码...

示例图片

声明:本文仅代表作者观点,不代表本站立场。如果侵犯到您的合法权益,请联系我们删除侵权资源!如果遇到资源链接失效,请您通过评论或工单的方式通知管理员。未经允许,不得转载,本站所有资源文章禁止商业使用运营!
下载安装【程序员客栈】APP
实时对接需求、及时收发消息、丰富的开放项目需求、随时随地查看项目状态

评论