use path

rasbt · May 12, 2024 · 33dda48 · 33dda48
1 parent 188d3cd
commit 33dda48
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ch06/03_bonus_imdb-classification/train-bert-hf.py b/ch06/03_bonus_imdb-classification/train-bert-hf.py
@@ -236,7 +236,7 @@ def train_classifier_simple(model, train_loader, val_loader, optimizer, device,
 
     pad_token_id = tokenizer.encode(tokenizer.pad_token)
 
-    base_path = "."
+    base_path = Path(".")
 
     train_dataset = IMDBDataset(base_path / "train.csv", max_length=256, tokenizer=tokenizer, pad_token_id=pad_token_id)
     val_dataset = IMDBDataset(base_path / "val.csv", max_length=256, tokenizer=tokenizer, pad_token_id=pad_token_id)