浏览代码

resolve conflict

/develop/bisim-review
Andrew Cohen 4 年前
当前提交
617aefc9
共有 1 个文件被更改,包括 18 次插入18 次删除
  1. 36
      ml-agents/mlagents/trainers/tests/test_simple_transfer.py

36
ml-agents/mlagents/trainers/tests/test_simple_transfer.py


# use_inverse_model=True
)
config = attr.evolve(
config, hyperparameters=new_hyperparams, max_steps=20000, summary_freq=5000
config, hyperparameters=new_hyperparams, max_steps=500000, summary_freq=5000
)
_check_environment_trains(
env, {BRAIN_NAME: config}, run_id=run_id + "_s" + str(seed), seed=seed

use_bisim=True,
)
config = attr.evolve(
config, hyperparameters=new_hyperparams, max_steps=20000, summary_freq=5000
config, hyperparameters=new_hyperparams, max_steps=500000, summary_freq=5000
)
_check_environment_trains(
env, {BRAIN_NAME: config}, run_id=run_id + "_s" + str(seed), seed=seed

if __name__ == "__main__":
for seed in range(5):
for obs in ["normal", "rich1", "rich2"]:
test_2d_model(seed=seed, obs_spec_type=obs, run_id="model_" + obs)
test_2d_model(seed=seed, obs_spec_type=obs, run_id="tmodel_" + obs)
for obs in ["normal", "rich2"]:
test_2d_transfer(
seed=seed,
obs_spec_type="rich1",
transfer_from="./transfer_results/model_" + obs + "_s" + str(seed) + "/Simple",
run_id=obs + "transfer_to_rich1",
)
for obs in ["normal", "rich1"]:
test_2d_transfer(
seed=seed,
obs_spec_type="rich2",
transfer_from="./transfer_results/model_" + obs + "_s" + str(seed) + "/Simple",
run_id=obs + "transfer_to_rich2",
)
# for obs in ["normal", "rich2"]:
# test_2d_transfer(
# seed=seed,
# obs_spec_type="rich1",
# transfer_from="./transfer_results/model_" + obs + "_s" + str(seed) + "/Simple",
# run_id=obs + "transfer_to_rich1",
# )
#
# for obs in ["normal", "rich1"]:
# test_2d_transfer(
# seed=seed,
# obs_spec_type="rich2",
# transfer_from="./transfer_results/model_" + obs + "_s" + str(seed) + "/Simple",
# run_id=obs + "transfer_to_rich2",
# )
# for obs in ["normal"]:

正在加载...
取消
保存