projects
/
culture.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
|
inline
| side by side (parent:
db8c213
)
Update.
author
François Fleuret
<francois@fleuret.org>
Sun, 30 Jun 2024 10:46:17 +0000
(13:46 +0300)
committer
François Fleuret
<francois@fleuret.org>
Sun, 30 Jun 2024 10:46:17 +0000
(13:46 +0300)
main.py
patch
|
blob
|
history
diff --git
a/main.py
b/main.py
index
a6c482f
..
6e5545c
100755
(executable)
--- 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("--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()
######################################################################
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":
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:
elif args.problem == "wireworld":
problem = wireworld.Wireworld(height=8, width=10, nb_iterations=2, speed=5)
else: