X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=main.py;h=e855b06c7461720bd420b71aad1705aba3bd3ee4;hb=62ad2378c60cdf322c0111279bd45fbef8365fc2;hp=2edfa14de0107376a431632791564451e729c298;hpb=e39282eef52a7f5ab6654b999009127569b1b599;p=picoclvr.git

diff --git a/main.py b/main.py
index 2edfa14..e855b06 100755
--- 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,
     )