diff --git a/torch_training/multi_agent_inference.py b/torch_training/multi_agent_inference.py index 8c1cbd0ebd6cf74e4bf698899797eae72d71072f..2fbbe61f7fd61eed4d8589695a06da238091717a 100644 --- a/torch_training/multi_agent_inference.py +++ b/torch_training/multi_agent_inference.py @@ -88,7 +88,7 @@ action_prob = [0] * action_size agent_obs = [None] * env.get_num_agents() agent_next_obs = [None] * env.get_num_agents() agent = Agent(state_size, action_size, "FC", 0) -with path(torch_training.Nets, "avoid_checkpoint60000.pth") as file_in: +with path(torch_training.Nets, "avoid_checkpoint100.pth") as file_in: agent.qnetwork_local.load_state_dict(torch.load(file_in)) record_images = False