From f3f490def0be8a3ea2b9a0ac60f5bb33c5c45fb5 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Sun, 7 Jan 2024 21:48:22 +0100 Subject: [PATCH] Update. --- README.txt | 4 ++++ fridge | 7 +++++++ mygpt.py | 2 -- 3 files changed, 11 insertions(+), 2 deletions(-) create mode 100644 README.txt diff --git a/README.txt b/README.txt new file mode 100644 index 0000000..abf3490 --- /dev/null +++ b/README.txt @@ -0,0 +1,4 @@ + +./main.py --task=memory --model=4M-C --nb_epochs=5 + +./main.py --task=grid --model=37M-C diff --git a/fridge b/fridge index 5dd85dd..ac7f86c 100644 --- a/fridge +++ b/fridge @@ -74,3 +74,10 @@ def insert_flash_back(rec_V, V, rec_K, K, t0, t1, CL, proba): ###################################################################### + +###################################################################### + +2024 Jan 07 21:38:11 (from mygpt.py) + + # insert_flash_back(self.rec_V,V,self.rec_K,K,t0,t1,CL,proba=self.proba_flashback / CL,) + diff --git a/mygpt.py b/mygpt.py index eda8685..de69a75 100755 --- a/mygpt.py +++ b/mygpt.py @@ -585,8 +585,6 @@ class Caterpillar(nn.Module): self.rec_K[:, :, t0:t1] = next_K.flatten(2, 3) if self.training and self.proba_flashback > 0.0: - # insert_flash_back(self.rec_V,V,self.rec_K,K,t0,t1,CL,proba=self.proba_flashback / CL,) - # This piece of code makes the assumption that there is # nothing informative before t0, otherwise we'd have to # implement a cache for V and K too. This should not be -- 2.20.1