final submission

script.py

@@ -154,7 +154,7 @@ def generate_embeddings(metadata_file_path, root_dir):
 
     test_dataset = ImageDataset(metadata_df, local_filepath=root_dir)
 
-    loader = DataLoader(test_dataset, batch_size=2, 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(