Update.

[culture.git] / main.py

diff --git a/main.py b/main.py
index 5484f39..a954af6 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -13,7 +13,7 @@ from torch.nn import functional as F
 
 import ffutils
 import mygpt
-import sky, wireworld, quizz_machine
+import sky, reasoning, quizz_machine
 
 # world quizzes vs. culture quizzes
 
@@ -79,8 +79,6 @@ parser.add_argument("--dropout", type=float, default=0.1)
 
 parser.add_argument("--deterministic_synthesis", action="store_true", default=False)
 
-parser.add_argument("--bidirectional_validation", action="store_true", default=False)
-
 parser.add_argument("--problem", type=str, default="sky")
 
 parser.add_argument("--nb_gpts", type=int, default=5)
@@ -91,12 +89,14 @@ parser.add_argument("--max_to_validate", type=int, default=None)
 
 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("--generation_temperature", type=float, default=2.0)
 
 parser.add_argument("--deterministic_validation", action="store_true", default=False)
 
+parser.add_argument("--bidirectional_validation", action="store_true", default=False)
+
+parser.add_argument("--dirty_debug", action="store_true", default=False)
+
 ######################################################################
 
 parser.add_argument("--sky_height", type=int, default=6)
@@ -114,10 +114,10 @@ parser.add_argument("--sky_speed", type=int, default=3)
 args = parser.parse_args()
 
 if args.min_to_validate is None:
-    args.min_to_validate = args = nb_gpts - 1
+    args.min_to_validate = args.nb_gpts - 1
 
 if args.max_to_validate is None:
-    args.max_to_validate = args = nb_gpts - 1
+    args.max_to_validate = args.nb_gpts - 1
 
 if args.result_dir is None:
     args.result_dir = f"results_culture"
@@ -249,8 +249,8 @@ if args.problem == "sky":
         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)
+elif args.problem == "reasoning":
+    problem = reasoning.Reasoning(device=device)
 else:
     raise ValueError