浏览代码

added the pypiwin32 package (#1668)

* added the pypiwin32 package

* fixed the break on mac, fixed part of pytest above version 4

* added something to the windows to help unstuck people

* resolved the comment
/develop-generalizationTraining-TrainerController
GitHub 6 年前
当前提交
cd087609
共有 3 个文件被更改,包括 12 次插入3 次删除
  1. 8
      docs/Installation-Windows.md
  2. 3
      ml-agents/setup.py
  3. 4
      ml-agents/tests/trainers/test_ppo.py

8
docs/Installation-Windows.md


This will complete the installation of all the required Python packages to run
the ML-Agents toolkit.
Sometimes on Windows, when you use pip to install certain python packages, the pip will get stuck when trying to read the cache of the package. If you see this, you can try:
```sh
pip install -e . --no-cache-dir
```
This `--no-cache-dir` tells the pip to disable the cache.
## (Optional) Step 4: GPU Training using The ML-Agents Toolkit
GPU is not required for the ML-Agents toolkit and won't speed up the PPO

3
ml-agents/setup.py


'docopt',
'pyyaml',
'protobuf>=3.6,<3.7',
'grpcio>=1.11.0,<1.12.0'],
'grpcio>=1.11.0,<1.12.0',
'pypiwin32==223;platform_system=="Windows"'],
python_requires=">=3.6,<3.7",

4
ml-agents/tests/trainers/test_ppo.py


@mock.patch('mlagents.envs.UnityEnvironment.executable_launcher')
@mock.patch('mlagents.envs.UnityEnvironment.get_communicator')
def test_ppo_policy_evaluate(mock_communicator, mock_launcher):
def test_ppo_policy_evaluate(mock_communicator, mock_launcher, dummy_config):
tf.reset_default_graph()
mock_communicator.return_value = MockCommunicator(
discrete_action=False, visual_inputs=0)

trainer_parameters = dummy_config()
trainer_parameters = dummy_config
model_path = env.brain_names[0]
trainer_parameters['model_path'] = model_path
trainer_parameters['keep_checkpoints'] = 3

正在加载...
取消
保存