create-caption

Paused

nroggendorff commited on Nov 16

Commit

4b8a72e

verified ·

1 Parent(s): 0186167

Update train.py

Files changed (1) hide show

train.py CHANGED Viewed

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

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