Skip to content

Latest commit

 

History

History
20 lines (12 loc) · 719 Bytes

README.md

File metadata and controls

20 lines (12 loc) · 719 Bytes
Raw

AnimalAI + Stable Baselines 3

A template repository for integrating the AnimalAI environment with Stable Baselines 3 agents.

This project was configured using Rye, but it should work with any tool supporting pyproject.toml.

Note: Check the version requirements, they are quite particular.
Note: Confirmed working with python == 3.9
Note: Very modern GPUs might not be compatible because of the older Torch version.

Usage

  • Playing rye run play
  • Training rye run train

Check the [project.scripts] section of pyproject.toml for the relevant configuration.

Resources

  • Docs for Stable Baselines 3 ==v2.0.0.