Tom Aarsen commited on
Commit
b312d97
·
1 Parent(s): dfd0c04

Clarify why we call quit()

Browse files
Files changed (1) hide show
  1. train.py +1 -0
train.py CHANGED
@@ -159,6 +159,7 @@ def load_train_eval_datasets():
159
  eval_dataset.save_to_disk("datasets/eval_dataset")
160
 
161
  # The `train_test_split` calls have put a lot of the datasets in memory, while we want it to just be on disk
 
162
  quit()
163
 
164
 
 
159
  eval_dataset.save_to_disk("datasets/eval_dataset")
160
 
161
  # The `train_test_split` calls have put a lot of the datasets in memory, while we want it to just be on disk
162
+ # So we're calling quit() here. Running the script again will load the datasets from disk.
163
  quit()
164
 
165