Skip to content
Snippets Groups Projects
Commit 7a4ea75b authored by Jae Young Lee's avatar Jae Young Lee
Browse files

High-level policy trained for 1m steps with 3-hidden layers.

parent 77d61b11
No related branches found
No related tags found
No related merge requests found
No preview for this file type
...@@ -47,8 +47,8 @@ def high_level_policy_training(nb_steps=25000, ...@@ -47,8 +47,8 @@ def high_level_policy_training(nb_steps=25000,
nb_actions=options.get_number_of_nodes(), nb_actions=options.get_number_of_nodes(),
target_model_update=1e-3, target_model_update=1e-3,
delta_clip=100, delta_clip=100,
low_level_policies=options.maneuvers, low_level_policies=options.maneuvers)
gamma=1) #gamma=1)
if load_weights: if load_weights:
agent.load_model(save_path) agent.load_model(save_path)
...@@ -78,8 +78,7 @@ def high_level_policy_testing(nb_episodes_for_test=100, ...@@ -78,8 +78,7 @@ def high_level_policy_testing(nb_episodes_for_test=100,
agent = DQNLearner( agent = DQNLearner(
input_shape=(50, ), input_shape=(50, ),
nb_actions=options.get_number_of_nodes(), nb_actions=options.get_number_of_nodes(),
low_level_policies=options.maneuvers, low_level_policies=options.maneuvers)
gamma=1)
if pretrained: if pretrained:
trained_agent_file = "backends/trained_policies/highlevel/" + trained_agent_file trained_agent_file = "backends/trained_policies/highlevel/" + trained_agent_file
...@@ -101,8 +100,7 @@ def evaluate_high_level_policy(nb_episodes_for_test=100, ...@@ -101,8 +100,7 @@ def evaluate_high_level_policy(nb_episodes_for_test=100,
agent = DQNLearner( agent = DQNLearner(
input_shape=(50, ), input_shape=(50, ),
nb_actions=options.get_number_of_nodes(), nb_actions=options.get_number_of_nodes(),
low_level_policies=options.maneuvers, low_level_policies=options.maneuvers)
gamma=1)
if pretrained: if pretrained:
trained_agent_file = "backends/trained_policies/highlevel/" + trained_agent_file trained_agent_file = "backends/trained_policies/highlevel/" + trained_agent_file
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment