浏览代码

Merge pull request #2472 from Unity-Technologies/release-0.9.2-multi-gpu-doc

Added the doc for multi-gpu
/develop-gpu-test
GitHub 5 年前
当前提交
df0196f9
共有 2 个文件被更改,包括 2 次插入1 次删除
  1. 1
      docs/Training-ML-Agents.md
  2. 2
      ml-agents/mlagents/trainers/learn.py

1
docs/Training-ML-Agents.md


[here](https://docs.unity3d.com/Manual/CommandLineArguments.html) for more
details.
* `--debug`: Specify this option to enable debug-level logging for some parts of the code.
* `--multi-gpu`: Setting this flag enables the use of multiple GPU's (if available) during training.
### Training Config File

2
ml-agents/mlagents/trainers/learn.py


--docker-target-name=<dt> Docker volume to store training-specific files [default: None].
--no-graphics Whether to run the environment in no-graphics mode [default: False].
--debug Whether to run ML-Agents in debug mode with detailed logging [default: False].
--multi-gpu Whether to use multiple GPU training [default: False].
--multi-gpu Setting this flag enables the use of multiple GPU's (if available) during training [default: False].
"""
options = docopt(_USAGE)

正在加载...
取消
保存