From: François Fleuret Date: Mon, 3 Jul 2023 09:27:44 +0000 (+0200) Subject: Update. X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=0b147af672d69d5fca328bc937467993c22fb20d;p=picoclvr.git Update. --- diff --git a/main.py b/main.py index c1f4dc7..e00ee66 100755 --- a/main.py +++ b/main.py @@ -922,7 +922,7 @@ class TaskStack(Task): i = torch.logical_and(self.test_input % 2 == 1, self.test_input < 2 * nb_stacks) counts = self.test_stack_counts.flatten()[i.flatten()] counts = F.one_hot(counts).sum(0) - log_string(f"pop_stack_counts {counts}") + log_string(f"test_pop_stack_counts {counts}") self.nb_codes = max(self.train_input.max(), self.test_input.max()) + 1