zeleznyt
/

snakeclef_test

Model card Files Files and versions Community

zeleznyt commited on May 24, 2024

Commit

ae2f470

·

verified ·

1 Parent(s): e16bca9

Update script.py

Files changed (1) hide show

script.py +1 -1

script.py CHANGED Viewed

@@ -85,7 +85,7 @@ if __name__ == "__main__":
         zip_ref.extractall("/tmp/data")
     # MODEL_PATH = "pytorch_model.bin"
-    MODEL_PATH = "2305_doublehead_best_loss.pth"
     # MODEL_NAME = "tf_efficientnet_b1.ap_in1k"
     MODEL_NAME = "swinv2_tiny_window16_256.ms_in1k"

         zip_ref.extractall("/tmp/data")
     # MODEL_PATH = "pytorch_model.bin"
+    MODEL_PATH = "2405_PSCmetric_3x_best_accuracy.pth"
     # MODEL_NAME = "tf_efficientnet_b1.ap_in1k"
     MODEL_NAME = "swinv2_tiny_window16_256.ms_in1k"