increase sample size

2024-05-19 15:38:41 +01:00
parent c8d8ec9928
commit 647ab70b46
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@@ -39,7 +39,7 @@ def load_data():
        trust_remote_code=True,
        split="train",
    )
-    flickr_dataset = load_dataset("nlphuji/flickr30k", split="test[:50%]")
+    flickr_dataset = load_dataset("nlphuji/flickr30k", split="test")
    painting_dataset = load_dataset(
        "keremberke/painting-style-classification", name="full", split="train"
    )