ML Agents don't move while training

Version information:
ml-agents: 0.21.0,
ml-agents-envs: 0.21.0,
Communicator API: 1.2.0,
TensorFlow: 2.3.1

StreamExecutor device (0): GeForce RTX 2080 Ti, Compute Capability 7.5

Hi, I’m not sure what the issue is here. Check out this video -

The agents move when Unity isn’t connected to mlagents-learn, but when I set it up to train they just do the first step and freeze. This is consistent across all the examples and mlagents-learn yaml files.

I’ve tried ml-agents releases 10, 11, and 12 with Unity 2018.4.30f1, 2019.4,18f1, and 2020.1.13f1. Same behavior across all versions, works fine with pre-trained inference, does not work when I try to us mlagents-learn. I’ve let it sit overnight and produce a 50,000,000 iteration file (toward end of video) The agents appear to move, but the training is bad so they fail.


I’ve been googling and watching videos of people doing the exact steps I’m doing, and the agents actually move and these values increase.

Followed the steps on our OSX build machine, works as expected.

Got it working -

I noticed on OSX it was saying “framework: pytorch” and on windows it said “framework: tensorflow”

I installed Anaconda and made a clean environment with these instructions →

Except when it throws the error about numpy I reinstalled mlagents with “pip install mlagents --use-feature=2020-resolver” and now the trainer works as expected.