Commit 418be8f0 authored by Unknown's avatar Unknown

20 for high level cost weifghts

parent 3d75ed5b
......@@ -152,6 +152,8 @@ class OptionsGraph:
lr=1e-3)
agent.load_model("backends/trained_policies/" + key + "/" + key + "_weights.h5f")
maneuver.set_low_level_trained_policy(agent.predict)
maneuver._cost_weights = (20.0 * 1e-3, 1.0 * 1e-3, 0.25 * 1e-3, 1.0 * 1e-3,
100.0 * 1e-3, 0.1 * 1e-3, 0.25 * 1e-3, 0.1 * 1e-3)
if self.config["method"] == "mcts":
maneuver.timeout = np.inf
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment