Whoops

bad_gpt.py

@@ -82,7 +82,7 @@ def estimate_loss(gpt: BadGPTModel, batcher: Batcher, eval_interval: int, device
         losses = torch.zeros(eval_interval)
         for epoch in range(eval_interval):
             train, answer = batcher.get_batch(split='train')
-            logits = gpt.forward(train)
+            logits = gpt(train)
             # Reformat pediction and answer so each entry can be compared
             batch, block, vocab = logits.shape
             logits = logits.view(batch * block, vocab)