sieberm
/

class_boost

Model card Files Files and versions Community

sieberm commited on May 24

Commit

f04df63

•

1 Parent(s): 13e1cce

Update script.py

Files changed (1) hide show

script.py +2 -2

script.py CHANGED Viewed

@@ -84,8 +84,8 @@ if __name__ == "__main__":
         zip_ref.extractall("/tmp/data")
     # MODEL_PATH = "best_accuracy_boost.pth"
-    MODEL_PATH= "best_loss_boost.pth"
-    # MODEL_PATH = "epoch_50.pth"
     MODEL_NAME = "swinv2_tiny_window16_256.ms_in1k"
     metadata_file_path = "./SnakeCLEF2024-TestMetadata.csv"

         zip_ref.extractall("/tmp/data")
     # MODEL_PATH = "best_accuracy_boost.pth"
+    #MODEL_PATH= "best_loss_boost.pth"
+    MODEL_PATH = "epoch_50.pth"
     MODEL_NAME = "swinv2_tiny_window16_256.ms_in1k"
     metadata_file_path = "./SnakeCLEF2024-TestMetadata.csv"