浏览代码

update config to train 5M steps

/hh-develop-pushblockcollab
HH 4 年前
当前提交
dd1fbd8a
共有 2 个文件被更改,包括 3 次插入2 次删除
  1. 3
      Project/Assets/ML-Agents/Examples/PushBlock/Scripts/PushBlockEnvController.cs
  2. 2
      config/ppo/PushBlock.yaml

3
Project/Assets/ML-Agents/Examples/PushBlock/Scripts/PushBlockEnvController.cs


m_ResetTimer += 1;
if (m_ResetTimer > MaxEnvironmentSteps)
{
m_ResetTimer = 0;
ResetScene();
}
}

void ResetScene()
{
m_ResetTimer = 0;
//Random platform rot
var rotation = Random.Range(0, 4);
var rotationAngle = rotation * 90f;

2
config/ppo/PushBlock.yaml


gamma: 0.99
strength: 1.0
keep_checkpoints: 5
max_steps: 2000000
max_steps: 5000000 #2000000
time_horizon: 64
summary_freq: 60000
threaded: true
正在加载...
取消
保存