From f480adc898e61e73c2835e1e2ca23cad85412092 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Sun, 4 Aug 2024 10:51:54 +0200 Subject: [PATCH] Update. --- main.py | 1 + 1 file changed, 1 insertion(+) diff --git a/main.py b/main.py index 55615ee..4dca41f 100755 --- a/main.py +++ b/main.py @@ -364,6 +364,7 @@ def sigma_for_grids(input): l = input.size(1) // 4 - 1 sigma = input.new(input.size()) r = sigma.view(sigma.size(0), 4, sigma.size(1) // 4) + r[:, :, 0] = 0 r[:, :, 1:] = ( torch.rand(input.size(0), 4, l, device=input.device).sort(dim=2).indices ) + 1 -- 2.39.5