Update.

diff --git a/grids.py b/grids.py
index 4c132c3..d319b9a 100755 (executable)
--- a/grids.py
+++ b/grids.py
@@ -287,8 +287,6 @@ class Grids(problem.Problem):
     ######################################################################
 
     def vocabulary_size(self):
-        # warnings.warn("hack +4 to keep the vocabulary size unchanged", RuntimeWarning)
-        # return self.nb_colors + 4
         return self.nb_colors
 
     def grid2img(self, x, scale=15, grids=True):

diff --git a/main.py b/main.py
index 4fee0f2..9e8589a 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -47,6 +47,7 @@ parser.add_argument("--eval_batch_size", type=int, default=25)
 
 parser.add_argument("--nb_train_samples", type=int, default=50000)
 
+
 parser.add_argument("--nb_test_samples", type=int, default=10000)
 
 parser.add_argument("--nb_c_quizzes", type=int, default=5000)
@@ -602,7 +603,8 @@ def one_complete_epoch(
 
     one_epoch(model, n_epoch, test_c_quizzes, train=False, local_device=local_device)
 
-    # Compute the test accuracy
+    # Compute the test accuracy *WITHOUT C QUIZZES*
+    warnings.warn("test_accuracy does not consider c quizzes", RuntimeWarning)
 
     quizzes = generate_quiz_set(
         args.nb_test_samples,