increase batch size

script.py

@@ -159,7 +159,7 @@ def generate_embeddings(metadata_file_path, root_dir):
 
     test_dataset = ImageDataset(metadata_df, local_filepath=root_dir)
 
-    loader = DataLoader(test_dataset, batch_size=1, shuffle=False)
+    loader = DataLoader(test_dataset, batch_size=3, shuffle=False)
 
     device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
     model = timm.create_model(