From 0b147af672d69d5fca328bc937467993c22fb20d Mon Sep 17 00:00:00 2001
From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= <francois@fleuret.org>
Date: Mon, 3 Jul 2023 11:27:44 +0200
Subject: [PATCH] Update.
---
main.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
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
--
2.39.5