From: François Fleuret Date: Mon, 23 Sep 2024 06:28:26 +0000 (+0200) Subject: Update. X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=f5e5ded2ac41dbd34e817aea469fa989e86e7acc;p=culture.git Update. --- 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,