diff --git a/torch_training/multi_agent_training.py b/torch_training/multi_agent_training.py index c9fe73308e20a113e5d393b4a16410f7611edb2b..5224ab68ac93905a5cf3cc8f436e9178e0523708 100644 --- a/torch_training/multi_agent_training.py +++ b/torch_training/multi_agent_training.py @@ -89,7 +89,7 @@ def main(argv): with path(torch_training.Nets, "avoid_checkpoint30000.pth") as file_in: agent.qnetwork_local.load_state_dict(torch.load(file_in)) - demo = True + demo = False record_images = False frame_step = 0