From 9897ed5294fc5f5ebe55f16a8da05474cbc8bf5b Mon Sep 17 00:00:00 2001
From: Guillaume Mollard <guillaume@iccluster091.iccluster.epfl.ch>
Date: Tue, 14 May 2019 13:51:48 +0200
Subject: [PATCH] fix in grid search, need to add gpu utilization
---
grid_search_configs/n_agents_grid_search/config.gin | 1 +
grid_search_train.py | 2 +-
2 files changed, 2 insertions(+), 1 deletion(-)
diff --git a/grid_search_configs/n_agents_grid_search/config.gin b/grid_search_configs/n_agents_grid_search/config.gin
index 430ba45..71035d4 100644
--- a/grid_search_configs/n_agents_grid_search/config.gin
+++ b/grid_search_configs/n_agents_grid_search/config.gin
@@ -1,5 +1,6 @@
run_grid_search.name = "n_agents_results"
run_grid_search.num_iterations = 1002
+run_grid_search.save_every = 200
run_grid_search.hidden_sizes = [32, 32]
run_grid_search.map_width = 15
diff --git a/grid_search_train.py b/grid_search_train.py
index 808ba9f..16b0a42 100644
--- a/grid_search_train.py
+++ b/grid_search_train.py
@@ -119,6 +119,6 @@ def run_grid_search(name, num_iterations, n_agents, hidden_sizes, save_every,
if __name__ == '__main__':
gin.external_configurable(tune.grid_search)
- dir = 'grid_search_configs/n_agents_grid_search'
+ dir = 'baselines/grid_search_configs/n_agents_grid_search'
gin.parse_config_file(dir + '/config.gin')
run_grid_search(local_dir=dir)
--
GitLab