浏览代码

fix commands for cloud training

/develop/bisim-review
yanchaosun 4 年前
当前提交
f0881a94
共有 3 个文件被更改,包括 4 次插入2 次删除
  1. 2
      config/ppo_transfer/CrawlerStatic.yaml
  2. 2
      ml-agents/mlagents/trainers/learn.py
  3. 2
      ml-agents/mlagents/trainers/policy/transfer_policy.py

2
config/ppo_transfer/CrawlerStatic.yaml


lambd: 0.95
num_epoch: 3
learning_rate_schedule: linear
encoder_layers: 2
policy_layers: 1
network_settings:
normalize: true
hidden_units: 512

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


run_seed = np.random.randint(0, 10000)
run_training(run_seed, options)
if options.behaviors.transfer:
os.system('mlagents-learn config/ppo_transfer/3DBallHard.yaml --env=/unity-volume/3dballhard --num-envs=4 --force')
os.system('mlagents-learn config/ppo_transfer/3DBallHard.yaml --run-id=hardball-transfer --env=/unity-volume/3dballhard --num-envs=4 --force')
def main():

2
ml-agents/mlagents/trainers/policy/transfer_policy.py


else:
predict_distribution = GaussianEncoderDistribution(
hidden_stream,
feature_sizex
feature_size
)
predict = predict_distribution.sample()

正在加载...
取消
保存