reduce diffusiondb sample size

2024-05-17 19:40:46 +01:00
parent 47b0b2e7a8
commit eaa6cf38ca
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@@ -33,7 +33,7 @@ def load_data():

    diffusion_db_dataset = load_dataset(
        "poloclub/diffusiondb",
-        "2m_random_50k",
+        "2m_random_10k",
        trust_remote_code=True,
        split="train",
    )