X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=mygpt.py;h=e3ff21f1420a0da492fbd34fe1ae96283ddcc57d;hb=b498eeca2b3abb2b4a370d4fdf39f172f2105bcf;hp=f10f1fe2013c4755a8ea7f2eef86a6b932059f25;hpb=6183291906184569c2206c34588d118cc77f74bb;p=mygptrnn.git diff --git a/mygpt.py b/mygpt.py index f10f1fe..e3ff21f 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 (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