X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=main.py;h=e855b06c7461720bd420b71aad1705aba3bd3ee4;hb=09952eb1ee41e279a1cb7797d2de997c6bcaa5af;hp=b4f831f8047dc4bf793a26c3f710b85844d318d9;hpb=22b841a39cc73310cd03dbd1d32fb387f68521d0;p=picoclvr.git

diff --git a/main.py b/main.py
index b4f831f..e855b06 100755
--- a/main.py
+++ b/main.py
@@ -178,11 +178,13 @@ 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=20)
+parser.add_argument("--escape_T", type=int, default=25)
+
+parser.add_argument("--escape_nb_walls", type=int, default=5)
 
 ######################################################################
 
@@ -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,
     )