##############################
# 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)
######################################################################
height=args.escape_height,
width=args.escape_width,
T=args.escape_T,
+ nb_walls=args.escape_nb_walls,
logger=log_string,
device=device,
)