WarpDrive + PytorchLightning trainer

salesforce / warp-drive

Extremely Fast End-to-End Deep Multi-Agent Reinforcement Learning Framework on a GPU (JMLR 2022)

BSD 3-Clause "New" or "Revised" License

465 stars 78 forks source link

Closed sunil-s closed 2 years ago

sunil-s commented 2 years ago

What this PR contains

WarpDrive trainer model that integrates with Pytorch Lightning
A tutorial on performing end-to-end RL training with WarpDrive and Pytorch Lightning