Create a log folder in enjoy.py

ikostrikov · Oct 8, 2017 · 89caf2a · 89caf2a
1 parent e6abe78
commit 89caf2a
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/enjoy.py b/enjoy.py
@@ -24,6 +24,11 @@
                     help='directory to save agent logs (default: /tmp/gym)')
 args = parser.parse_args()
 
+try:
+    os.makedirs(args.log_dir)
+except OSError:
+    pass
+
 env = make_env(args.env_name, args.seed, 0, args.log_dir)()
 
 actor_critic = torch.load(os.path.join(args.load_dir, args.env_name + ".pt"))