From 647ab70b4690b1e8224f475057b62d7b278ece5b Mon Sep 17 00:00:00 2001 From: Kenneth Date: Sun, 19 May 2024 15:38:41 +0100 Subject: [PATCH] increase sample size --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index d1caa7d..a412735 100644 --- 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" )