强化AI学习 NVIDIA新作将训练速度大幅提升
2021-09-06 22:22 来源:未知
英伟达NVIDIA为研究人员开辟了Isaac Gym高机能机械人模仿平台。它可以运行一个端到端的GPU加快练习管道,使研究人员可以或许克服诸多限制,在持续控制义务中实现100倍-1000倍的练习速度。
Isaac Gym经由过程将物理缓存中的数据直接传递到 PyTorch 张量进行通信,可以在 GPU 上端到端地实现物理模仿和神经收集策略练习。 Isaac Gym 供给了一个高机能的进修平台,可以直接在 GPU 长进行各类代理练习。
与应用 CPU 模仿器和 GPU 神经收集的传统 RL 练习比拟,Isaac Gym 大年夜大年夜削减了单个 GPU 上复杂机械义务的练习时光,其练习速度进步了 1-2 个数量级。