fix bug free only if initted

staar · May 23, 2024 · d3cf9e2 · d3cf9e2
1 parent eda0c2f
commit d3cf9e2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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);