From ad41c6e3cc3d97a5ea84eebd6031c905c0248bfd Mon Sep 17 00:00:00 2001 From: rasbt Date: Sun, 12 May 2024 09:41:46 -0500 Subject: [PATCH] use validation 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 e3da2b1..8d9c796 100644 --- a/ch06/03_bonus_imdb-classification/train-bert-hf.py +++ b/ch06/03_bonus_imdb-classification/train-bert-hf.py @@ -239,7 +239,7 @@ if __name__ == "__main__": 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) + val_dataset = IMDBDataset(base_path / "validation.csv", max_length=256, tokenizer=tokenizer, pad_token_id=pad_token_id) test_dataset = IMDBDataset(base_path / "test.csv", max_length=256, tokenizer=tokenizer, pad_token_id=pad_token_id) num_workers = 0