From: François Fleuret Date: Sat, 6 Jul 2024 05:28:12 +0000 (+0300) Subject: Update. X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=f5cce89b62807db901f63783ce3f329758480c6c;p=culture.git Update. --- 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)