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