Update.

diff --git a/reasoning.py b/reasoning.py
index 951e04a..fb208b0 100755 (executable)
--- a/reasoning.py
+++ b/reasoning.py
@@ -117,7 +117,7 @@ class Reasoning(problem.Problem):
                 c = c.long()[:, None]
                 c = (
                     (1 - ((c == 1).long() + (c == 0).long() + (c == -1).long()))
-                    * torch.tensor([128, 128, 128], device=c.device)
+                    * torch.tensor([64, 64, 64], device=c.device)
                     + (c == 1).long() * torch.tensor([0, 255, 0], device=c.device)
                     + (c == 0).long() * torch.tensor([255, 255, 255], device=c.device)
                     + (c == -1).long() * torch.tensor([255, 0, 0], device=c.device)