projects
/
culture.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
|
inline
| side by side (from parent 1:
f91736e
)
Update.
author
François Fleuret
<francois@fleuret.org>
Mon, 3 Jul 2023 09:27:44 +0000
(11:27 +0200)
committer
François Fleuret
<francois@fleuret.org>
Mon, 3 Jul 2023 09:27:44 +0000
(11:27 +0200)
main.py
patch
|
blob
|
history
diff --git
a/main.py
b/main.py
index
c1f4dc7
..
e00ee66
100755
(executable)
--- 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)
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
self.nb_codes = max(self.train_input.max(), self.test_input.max()) + 1