create-caption

Paused

nroggendorff commited on Nov 16

Commit

8e44691

verified ·

1 Parent(s): dae35de

Update train.py

Files changed (1) hide show

train.py CHANGED Viewed

@@ -130,7 +130,7 @@ def main():
     input_dataset = "none-yet/anime-captions"
     output_dataset = "nroggendorff/anime-captions"
     model_name = "datalab-to/chandra"
-    batch_size = 16
     print("Loading dataset info...")
     loaded = datasets.load_dataset(input_dataset, split="train")

     input_dataset = "none-yet/anime-captions"
     output_dataset = "nroggendorff/anime-captions"
     model_name = "datalab-to/chandra"
+    batch_size = 12
     print("Loading dataset info...")
     loaded = datasets.load_dataset(input_dataset, split="train")