zeleznyt commited on
Commit
c68f173
·
verified ·
1 Parent(s): 71f8226

Update script.py

Browse files
Files changed (1) hide show
  1. script.py +1 -3
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 = ["2405_PSCmetric2_0001_best_accuracy.pth",
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"