athenlab/mai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

increase sample size

Browse Source
This commit is contained in:
Kenneth
2024-05-19 15:38:41 +01:00
parent c8d8ec9928
commit 647ab70b46
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@@ -39,7 +39,7 @@ def load_data():
trust_remote_code=True, trust_remote_code=True,
split="train", split="train",
) )
flickr_dataset = load_dataset("nlphuji/flickr30k", split="test[:50%]") flickr_dataset = load_dataset("nlphuji/flickr30k", split="test")
painting_dataset = load_dataset( painting_dataset = load_dataset(
"keremberke/painting-style-classification", name="full", split="train" "keremberke/painting-style-classification", name="full", split="train"
) )