Update.

diff --git a/mygpt.py b/mygpt.py
index fb08a3a..24ba345 100755 (executable)
--- a/mygpt.py
+++ b/mygpt.py
@@ -695,8 +695,6 @@ class Caterpillar(nn.Module):
                 + (1 - mask_K) * self.rec_K[:, :, t0:t1]
             )
 
-        exit(0)
-
         ######################################################################
         # compute the readout