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

Updates to the training code #20

Closed sunil-s closed 2 years ago

sunil-s commented 2 years ago

added param scheduler, episodic returns, default config, trainer refactoring and formatting