From: François Fleuret Date: Sun, 7 Jan 2024 10:56:06 +0000 (+0100) Subject: Update. X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=b498eeca2b3abb2b4a370d4fdf39f172f2105bcf;p=mygptrnn.git Update. --- diff --git a/mygpt.py b/mygpt.py index 984ae67..e3ff21f 100755 --- a/mygpt.py +++ b/mygpt.py @@ -472,9 +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) - # 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 + # Set a origin source time (starting time of the chunck to copy + # here) We set it as the current time minus 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