Commit ac9c8016 authored by Erik Nygren's avatar Erik Nygren
Browse files

first tests with new tree search observation. looks good :)

parent f1b63471
Pipeline #480 failed with stage
in 1 minute and 48 seconds
...@@ -55,9 +55,9 @@ scores = [] ...@@ -55,9 +55,9 @@ scores = []
dones_list = [] dones_list = []
action_prob = [0] * 4 action_prob = [0] * 4
agent = Agent(state_size, action_size, "FC", 0) agent = Agent(state_size, action_size, "FC", 0)
#agent.qnetwork_local.load_state_dict(torch.load('../flatland/baselines/Nets/avoid_checkpoint14900.pth')) agent.qnetwork_local.load_state_dict(torch.load('../flatland/baselines/Nets/avoid_checkpoint14900.pth'))
demo = False demo = True
def max_lt(seq, val): def max_lt(seq, val):
......
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