updated training and inference to flatland 2.0 level generators

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