last try for real

script.py

@@ -117,6 +117,7 @@ class ImageDataset(Dataset):
 
     def __getitem__(self, idx):
         image_path = os.path.join(self.local_filepath, self.filepaths[idx])
+        # print("Reading from ", image_path)
 
         image = cv2.imread(image_path)
         image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
@@ -158,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, num_workers=4)
+    loader = DataLoader(test_dataset, batch_size=1, shuffle=False)
 
     device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
     model = timm.create_model(
@@ -335,8 +336,7 @@ def make_submission(metadata_df):
             {w: ckpt["model." + w] for w in model.state_dict().keys()}
         )
         model.eval()
-        model.cuda()
-
+        
         models.append(model)
 
     fungi_model = FungiEnsembleModel(models)
@@ -396,3 +396,7 @@ if __name__ == "__main__":
     metadata_df = generate_embeddings(metadata_file_path, root_dir)
 
     make_submission(metadata_df)
+
+
+
+