From 25b81a9ba8f5526151182640edaf797921860b40 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Sun, 7 Jan 2024 11:55:39 +0100 Subject: [PATCH] Update. --- mygpt.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/mygpt.py b/mygpt.py index f10f1fe..984ae67 100755 --- a/mygpt.py +++ b/mygpt.py @@ -472,8 +472,9 @@ def flash_back_time_src(N, H, t0, t1, CL, CH, proba, device): fb_body = fb_body.cumsum(dim=2) fb_start = fb_start * (fb_body == 1) - # t_s = t0-(t0//L * R)*L - + # Set a origin source time (at what time start the chunck to copy + # here) We set it as a multiple of CL to be consistent with the + # "rolling" caterpillar t = torch.arange(fb_start.size(2), device=fb_start.device)[None, None, :] src_time = fb_start * ( t -- 2.20.1