Update script.py
Browse files
script.py
CHANGED
@@ -83,8 +83,8 @@ if __name__ == "__main__":
|
|
83 |
with zipfile.ZipFile("/tmp/data/private_testset.zip", 'r') as zip_ref:
|
84 |
zip_ref.extractall("/tmp/data")
|
85 |
|
86 |
-
MODEL_PATH = "best_accuracy_boost.pth"
|
87 |
-
|
88 |
# MODEL_PATH = "epoch_50.pth"
|
89 |
MODEL_NAME = "swinv2_tiny_window16_256.ms_in1k"
|
90 |
|
|
|
83 |
with zipfile.ZipFile("/tmp/data/private_testset.zip", 'r') as zip_ref:
|
84 |
zip_ref.extractall("/tmp/data")
|
85 |
|
86 |
+
# MODEL_PATH = "best_accuracy_boost.pth"
|
87 |
+
MODEL_PATH= "best_loss_boost.pth"
|
88 |
# MODEL_PATH = "epoch_50.pth"
|
89 |
MODEL_NAME = "swinv2_tiny_window16_256.ms_in1k"
|
90 |
|