From 3c6931f8ddc8160550e026d9e9610ef71260ce10 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Sun, 7 Jan 2024 16:23:49 +0100 Subject: [PATCH] Update. --- mygpt.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mygpt.py b/mygpt.py index 5ea927e..b5ac916 100755 --- a/mygpt.py +++ b/mygpt.py @@ -655,7 +655,7 @@ class Caterpillar(nn.Module): self.rec_V[:, :, t0:t1] = next_V.flatten(2, 3) self.rec_K[:, :, t0:t1] = next_K.flatten(2, 3) - if self.training and self.proba_flashback: + 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 -- 2.39.5