From f5e5ded2ac41dbd34e817aea469fa989e86e7acc Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Mon, 23 Sep 2024 08:28:26 +0200 Subject: [PATCH] Update. --- grids.py | 2 -- main.py | 4 +++- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/grids.py b/grids.py index 4c132c3..d319b9a 100755 --- 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 --- 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, -- 2.39.5