From: François Fleuret Date: Sun, 30 Jun 2024 10:46:17 +0000 (+0300) Subject: Update. X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=9ff76e2a501ec296fbb8f6c2d12ad4ceac148b5f;p=culture.git Update. --- diff --git a/main.py b/main.py index a6c482f..6e5545c 100755 --- a/main.py +++ b/main.py @@ -95,6 +95,16 @@ parser.add_argument("--accuracy_to_make_c_quizzes", type=float, default=0.975) parser.add_argument("--dirty_debug", action="store_true", default=False) +parser.add_argument("--sky_height", type=int, default=6) + +parser.add_argument("--sky_width", type=int, default=8) + +parser.add_argument("--sky_nb_birds", type=int, default=3) + +parser.add_argument("--sky_nb_iterations", type=int, default=2) + +parser.add_argument("--sky_speed", type=int, default=3) + ###################################################################### args = parser.parse_args() @@ -222,7 +232,13 @@ assert args.nb_train_samples % args.batch_size == 0 assert args.nb_test_samples % args.batch_size == 0 if args.problem == "sky": - problem = sky.Sky(height=6, width=8, nb_birds=3, nb_iterations=2, speed=3) + problem = sky.Sky( + height=args.sky_height, + width=args.sky_width, + nb_birds=args.sky_nb_birds, + nb_iterations=args.sky_nb_iterations, + speed=args.sky_speed, + ) elif args.problem == "wireworld": problem = wireworld.Wireworld(height=8, width=10, nb_iterations=2, speed=5) else: