From d3cf9e211c2a606371d8fab411eae5719097703d Mon Sep 17 00:00:00 2001
From: Andrej Karpathy <andrej.karpathy@gmail.com>
Date: Thu, 23 May 2024 00:49:32 +0000
Subject: [PATCH] fix bug free only if initted

---
 train_gpt2.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train_gpt2.cu b/train_gpt2.cu
index 62e701a6f..798fe763b 100644
--- a/train_gpt2.cu
+++ b/train_gpt2.cu
@@ -2893,7 +2893,7 @@ int main(int argc, char *argv[]) {
     // free and destroy everything
     cudaCheck(cudaEventDestroy(end));
     cudaCheck(cudaEventDestroy(start));
-    evalloader_free(&eval_loader);
+    if (hellaswag_available) { evalloader_free(&eval_loader); }
     dataloader_free(&train_loader);
     dataloader_free(&val_loader);
     tokenizer_free(&tokenizer);