Update script.py
Browse files
script.py
CHANGED
@@ -85,7 +85,7 @@ if __name__ == "__main__":
|
|
85 |
zip_ref.extractall("/tmp/data")
|
86 |
|
87 |
# MODEL_PATH = "pytorch_model.bin"
|
88 |
-
MODEL_PATH = "
|
89 |
# MODEL_NAME = "tf_efficientnet_b1.ap_in1k"
|
90 |
MODEL_NAME = "swinv2_tiny_window16_256.ms_in1k"
|
91 |
|
|
|
85 |
zip_ref.extractall("/tmp/data")
|
86 |
|
87 |
# MODEL_PATH = "pytorch_model.bin"
|
88 |
+
MODEL_PATH = "best_accuracy_2205_2.pth"
|
89 |
# MODEL_NAME = "tf_efficientnet_b1.ap_in1k"
|
90 |
MODEL_NAME = "swinv2_tiny_window16_256.ms_in1k"
|
91 |
|