Update script.py

script.py

@@ -64,12 +64,7 @@ class PytorchWorker:
         _, preds = torch.max(outputs, 1)
         
         preds = preds.cpu()  # Move tensor to CPU    
-
-        # post process
-        max_value = torch.max(outputs)
-        if max_value < -10:
-            preds[0]=1604
-            
+        
         print("preds: ", preds)
 
         return preds.tolist()  # Convert tensor to list
@@ -108,8 +103,9 @@ if __name__ == "__main__":
 
     # MODEL_PATH = './efficientnet_b3_epoch_9_delete_pre.pth' # "./efficientnet_b3_epoch_9.pth"
     # MODEL_PATH = './efficientnet_b3_epoch_24_trick1.2.3_0.6067.pth'
-    MODEL_PATH = './efficientnet_b3_epoch_10_trick1.2.4_0.6016.pth'
+    # MODEL_PATH = './efficientnet_b3_epoch_10_trick1.2.4_0.6016.pth'
     # MODEL_PATH = './efficientnet_b3_epoch_3_trick1.2.3_a0.6067_l5.6311.pth'
+    MODEL_PATH = './efficientnet_b3_epoch_21_trick1.2.5_a0.7237_l17.1662.pth'
     MODEL_NAME = 'tf_efficientnet_b3_ns' #"tf_efficientnet_b1.ap_in1k"
 
     metadata_file_path = "./FungiCLEF2024_TestMetadata.csv"