Skip to content
Snippets Groups Projects
Commit da9d805a authored by u214892's avatar u214892
Browse files

#42 run baselines in ci

parent 622cab24
No related branches found
No related tags found
No related merge requests found
...@@ -82,6 +82,12 @@ def main(argv): ...@@ -82,6 +82,12 @@ def main(argv):
agent_obs = [None] * env.get_num_agents() agent_obs = [None] * env.get_num_agents()
agent_next_obs = [None] * env.get_num_agents() agent_next_obs = [None] * env.get_num_agents()
agent = Agent(state_size, action_size, "FC", 0) agent = Agent(state_size, action_size, "FC", 0)
with path(torch_training.Nets, "avoid_checkpoint30000.pth") as file_in:
agent.qnetwork_local.load_state_dict(torch.load(file_in))
demo = False
record_images = False
frame_step = 0
print("multi_agent_trainging.py (2)") print("multi_agent_trainging.py (2)")
......
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