reduce batch size

script.py

@@ -66,7 +66,7 @@ def generate_embeddings(metadata_file_path, root_dir):
         metadata_df, local_filepath=root_dir, transform=transforms
     )
 
-    loader = DataLoader(test_dataset, batch_size=3, shuffle=False, num_workers=4)
+    loader = DataLoader(test_dataset, batch_size=1, shuffle=False, num_workers=4)
 
     device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
     model = timm.create_model("timm/vit_large_patch14_reg4_dinov2.lvd142m", pretrained=True)