Update.

[picoclvr.git] / main.py

diff --git a/main.py b/main.py
index 2edfa14..e855b06 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -178,12 +178,14 @@ parser.add_argument("--mixing_deterministic_start", action="store_true", default
 ##############################
 # escape options
 
-parser.add_argument("--escape_height", type=int, default=4)
+parser.add_argument("--escape_height", type=int, default=5)
 
-parser.add_argument("--escape_width", type=int, default=6)
+parser.add_argument("--escape_width", type=int, default=7)
 
 parser.add_argument("--escape_T", type=int, default=25)
 
+parser.add_argument("--escape_nb_walls", type=int, default=5)
+
 ######################################################################
 
 args = parser.parse_args()
@@ -622,6 +624,7 @@ elif args.task == "escape":
         height=args.escape_height,
         width=args.escape_width,
         T=args.escape_T,
+        nb_walls=args.escape_nb_walls,
         logger=log_string,
         device=device,
     )