From 33dda489a143b7c0a3edf968f1e993f5e7830fcb Mon Sep 17 00:00:00 2001 From: rasbt Date: Sun, 12 May 2024 09:36:35 -0500 Subject: [PATCH] use path --- ch06/03_bonus_imdb-classification/train-bert-hf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ch06/03_bonus_imdb-classification/train-bert-hf.py b/ch06/03_bonus_imdb-classification/train-bert-hf.py index aa207c9..e3da2b1 100644 --- a/ch06/03_bonus_imdb-classification/train-bert-hf.py +++ b/ch06/03_bonus_imdb-classification/train-bert-hf.py @@ -236,7 +236,7 @@ if __name__ == "__main__": 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)