Skip to content
Snippets Groups Projects
Commit 54224ea3 authored by Erik Nygren's avatar Erik Nygren
Browse files

updated to incoroporate new loading method

parent 1cb51539
No related branches found
No related tags found
No related merge requests found
...@@ -89,7 +89,7 @@ def main(argv): ...@@ -89,7 +89,7 @@ def main(argv):
with path(torch_training.Nets, "avoid_checkpoint30000.pth") as file_in: with path(torch_training.Nets, "avoid_checkpoint30000.pth") as file_in:
agent.qnetwork_local.load_state_dict(torch.load(file_in)) agent.qnetwork_local.load_state_dict(torch.load(file_in))
demo = True demo = False
record_images = False record_images = False
frame_step = 0 frame_step = 0
......
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