Update script.py
Browse files
script.py
CHANGED
@@ -93,9 +93,7 @@ if __name__ == "__main__":
|
|
93 |
zip_ref.extractall("/tmp/data")
|
94 |
|
95 |
# MODEL_PATH = "pytorch_model.bin"
|
96 |
-
MODEL_PATH = ["
|
97 |
-
"2405_PSCmetric_3x_best_accuracy.pth",
|
98 |
-
"2405_PSCmetric_real3_best_accuracy.pth",
|
99 |
"2405_cls_boost_best_accuracy.pth"
|
100 |
]
|
101 |
# MODEL_NAME = "tf_efficientnet_b1.ap_in1k"
|
|
|
93 |
zip_ref.extractall("/tmp/data")
|
94 |
|
95 |
# MODEL_PATH = "pytorch_model.bin"
|
96 |
+
MODEL_PATH = ["2405_PSCmetric_real3_best_accuracy.pth",
|
|
|
|
|
97 |
"2405_cls_boost_best_accuracy.pth"
|
98 |
]
|
99 |
# MODEL_NAME = "tf_efficientnet_b1.ap_in1k"
|