X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;ds=sidebyside;f=mygpt.py;h=7f0fb9b6fa506a5136ff4e98c8b9f5a4087420ee;hb=8a32cb4548bb48ef68adb4df9372fe5f7a80b67c;hp=984ae6703afa857e8bcdaa2e47678fd1604afbb9;hpb=25b81a9ba8f5526151182640edaf797921860b40;p=mygptrnn.git

diff --git a/mygpt.py b/mygpt.py
index 984ae67..7f0fb9b 100755
--- a/mygpt.py
+++ b/mygpt.py
@@ -457,7 +457,8 @@ def moving_window(x, dim, win_dim, win_size):
 
 ##############################
 
-# This is one order of magnitude more complicated than I expected
+# This is one order of magnitude more complicated than I expected, not
+# elegant, slow, hopefully not buggy
 
 
 def flash_back_time_src(N, H, t0, t1, CL, CH, proba, device):
@@ -472,9 +473,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