sieberm
/

class_boost

Model card Files Files and versions

sieberm commited on May 24, 2024

Commit

13e1cce

·

verified ·

1 Parent(s): 8941f8e

Update script.py

Files changed (1) hide show

script.py +2 -2

script.py CHANGED Viewed

@@ -83,8 +83,8 @@ if __name__ == "__main__":
     with zipfile.ZipFile("/tmp/data/private_testset.zip", 'r') as zip_ref:
         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"

     with zipfile.ZipFile("/tmp/data/private_testset.zip", 'r') as zip_ref:
         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"