From f04c2dfdbc506e7353811959473612dfd55bc42b Mon Sep 17 00:00:00 2001
From: Guillaume Mollard <guillaume.mollard2@gmail.com>
Date: Sat, 29 Jun 2019 15:15:08 +0200
Subject: [PATCH] small fixes
---
RLLib_training/render_training_result.py | 4 ++--
RLLib_training/train_experiment.py | 8 ++++----
2 files changed, 6 insertions(+), 6 deletions(-)
diff --git a/RLLib_training/render_training_result.py b/RLLib_training/render_training_result.py
index 90b38a3..668c763 100644
--- a/RLLib_training/render_training_result.py
+++ b/RLLib_training/render_training_result.py
@@ -32,7 +32,7 @@ ray.init() # object_store_memory=150000000000, redis_max_memory=30000000000)
__file_dirname__ = os.path.dirname(os.path.realpath(__file__))
CHECKPOINT_PATH = os.path.join(__file_dirname__, 'experiment_configs', 'config_example', 'ppo_policy_two_obs_with_predictions_n_agents_4_map_size_20q58l5_f7',
- 'checkpoint_101', 'checkpoint-101')
+ 'checkpoint_101', 'checkpoint-101') # To Modify
N_EPISODES = 10
N_STEPS_PER_EPISODE = 50
@@ -165,5 +165,5 @@ def run_experiment(name, num_iterations, n_agents, hidden_sizes, save_every,
if __name__ == '__main__':
- gin.parse_config_file(os.path.join(__file_dirname__, 'experiment_configs', 'config_example', 'config.gin'))
+ gin.parse_config_file(os.path.join(__file_dirname__, 'experiment_configs', 'config_example', 'config.gin')) # To Modify
run_experiment()
diff --git a/RLLib_training/train_experiment.py b/RLLib_training/train_experiment.py
index e7e9709..808bbc3 100644
--- a/RLLib_training/train_experiment.py
+++ b/RLLib_training/train_experiment.py
@@ -179,7 +179,7 @@ def run_experiment(name, num_iterations, n_agents, hidden_sizes, save_every,
"kl_coeff": kl_coeff,
"lambda_gae": lambda_gae,
"min_dist": min_dist,
- "step_memory": step_memory
+ "step_memory": step_memory
},
resources_per_trial={
"cpu": 3,
@@ -191,7 +191,7 @@ def run_experiment(name, num_iterations, n_agents, hidden_sizes, save_every,
if __name__ == '__main__':
- print(str(os.path.join(__file_dirname__, 'experiment_configs', 'config_example', 'config.gin')))
- gin.parse_config_file(os.path.join(__file_dirname__, 'experiment_configs', 'config_example', 'config.gin'))
- dir = os.path.join(__file_dirname__, 'experiment_configs', 'config_example')
+ folder_name = 'config_example' # To Modify
+ gin.parse_config_file(os.path.join(__file_dirname__, 'experiment_configs', folder_name, 'config.gin'))
+ dir = os.path.join(__file_dirname__, 'experiment_configs', folder_name)
run_experiment(local_dir=dir)
--
GitLab