g8a9 commited on
Commit
0835232
1 Parent(s): daa8529

Minor change

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -59,7 +59,7 @@ def get_image_features(model, image_dir):
59
 
60
  loader = torch.utils.data.DataLoader(
61
  dataset,
62
- batch_size=64,
63
  shuffle=False,
64
  num_workers=8,
65
  persistent_workers=True,
 
59
 
60
  loader = torch.utils.data.DataLoader(
61
  dataset,
62
+ batch_size=32,
63
  shuffle=False,
64
  num_workers=8,
65
  persistent_workers=True,