Update script.py

script.py

@@ -104,8 +104,8 @@ 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_22_trick1.3_0.5977.pth'
-    MODEL_NAME = 'tf_efficientnet_b7_ns' #"tf_efficientnet_b1.ap_in1k"
+    MODEL_PATH = './efficientnet_b3_epoch_3_trick1.2.3_a0.6067_l5.6311.pth'
+    MODEL_NAME = 'tf_efficientnet_b3_ns' #"tf_efficientnet_b1.ap_in1k"
 
     metadata_file_path = "./FungiCLEF2024_TestMetadata.csv"
     test_metadata = pd.read_csv(metadata_file_path)
@@ -114,4 +114,4 @@ if __name__ == "__main__":
         test_metadata=test_metadata,
         model_path=MODEL_PATH,
         model_name=MODEL_NAME
-    )
+    )