Spaces:

clip-italian
/

clip-italian-demo

Running

g8a9 commited on Jul 17, 2021

Commit

d00235e

1 Parent(s): 128a4c1

Change BS

Files changed (1) hide show

app.py CHANGED Viewed

@@ -59,9 +59,9 @@ def get_image_features(model, image_dir):
     loader = torch.utils.data.DataLoader(
         dataset,
-        batch_size=256,
         shuffle=False,
-        num_workers=2,
         persistent_workers=True,
         drop_last=False,
     )

     loader = torch.utils.data.DataLoader(
         dataset,
+        batch_size=64,
         shuffle=False,
+        num_workers=8,
         persistent_workers=True,
         drop_last=False,
     )