From f06a70eca52e988857ee043f1379d41b09dd365d Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Sat, 6 Jan 2024 14:44:50 +0100 Subject: [PATCH] Update. --- mygpt.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mygpt.py b/mygpt.py index 7105e97..9bacaff 100755 --- a/mygpt.py +++ b/mygpt.py @@ -442,7 +442,8 @@ class KVRec(nn.Module): # Returns a tensor with an additional index at rank win_dim, that move -# along the same dimension as dim, on a domain {0...win_size-1} +# along the same dimension as dim, on a domain {0...win_size-1}, and +# dim is restricted on a domain reduced by win_size-1 values. def moving_window(x, dim, win_dim, win_size): -- 2.20.1