GitHub
36ed3c16
Fix issue exporting graph with multi-GPU (#2573)
Our multi-GPU training had a regression such that freezing the graph was broken. This change fixes that issue by making a few changes: * Removes the top level "tower" variable scope added by multi-GPU so that the output nodes have correct names * Removes the use of "freeze_graph" and replaces it with our own similar functionality. * Adds the "auto reuse" to network layers which require them |
5 年前 | |
---|---|---|
.. | ||
mlagents/trainers | Fix issue exporting graph with multi-GPU (#2573) | 5 年前 |
README.md | Fixing tables in documentation and other markdown errors. (#1199) | 6 年前 |
setup.py | Use argparse for arg parsing (#2586) | 5 年前 |
README.md
Unity ML-Agents Python Interface and Trainers
The mlagents
Python package is part of the
ML-Agents Toolkit.
mlagents
provides a Python API that allows direct interaction with the Unity
game engine as well as a collection of trainers and algorithms to train agents
in Unity environments.
The mlagents
Python package contains two sub packages:
-
mlagents.envs
: A low level API which allows you to interact directly with a Unity Environment. See here for more information on using this package. -
mlagents.trainers
: A set of Reinforcement Learning algorithms designed to be used with Unity environments. Access them using the:mlagents-learn
access point. See here for more information on using this package.
Installation
Install the mlagents
package with:
pip install mlagents
Usage & More Information
For more detailed documentation, check out the ML-Agents Toolkit documentation.