From f5cce89b62807db901f63783ce3f329758480c6c Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Sat, 6 Jul 2024 08:28:12 +0300 Subject: [PATCH] Update. --- reasoning.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/reasoning.py b/reasoning.py index 951e04a..fb208b0 100755 --- 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) -- 2.20.1