basepath

rasbt · May 12, 2024 · a733a7e · a733a7e
1 parent 2e47a6e
commit a733a7e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ch06/03_bonus_imdb-classification/train-gpt.py b/ch06/03_bonus_imdb-classification/train-gpt.py
@@ -286,6 +286,8 @@ def train_classifier_simple(model, train_loader, val_loader, optimizer, device,
     # Instantiate dataloaders
     ###############################
 
+    base_path = Path(".")
+
     tokenizer = tiktoken.get_encoding("gpt2")
 
     train_dataset = None