浏览代码

Merging master

/develop/gym-wrapper
vincentpierre 4 年前
当前提交
d031c7a9
共有 1 个文件被更改,包括 1 次插入3 次删除
  1. 4
      ml-agents/mlagents/trainers/tests/test_simple_rl.py

4
ml-agents/mlagents/trainers/tests/test_simple_rl.py


)
env = GymToUnityWrapper(gym.make(gym_name), BRAIN_NAME)
hyperparams = attr.evolve(
SAC_CONFIG.hyperparameters,
learning_rate=3e-4,
buffer_size=1000, # , num_layers=1, normalize=True, time_horizon=256, hidden_units=64
SAC_CONFIG.hyperparameters, learning_rate=3e-4, buffer_size=1000
)
config = attr.evolve(SAC_CONFIG, hyperparameters=hyperparams, max_steps=50000)

正在加载...
取消
保存