j029

nmcardoso · Aug 23, 2023 · 834f417 · 834f417
1 parent 0ac643d
commit 834f417
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mergernet/jobs/j029.py b/mergernet/jobs/j029.py
@@ -26,7 +26,7 @@ def __init__(self):
 
 
   def call(self):
-    model = load_model('model.h5', exp_id=28)
+    model = load_model('model_softmax_logits.h5', exp_id=28)
     ds = Dataset(Dataset.registry.LS10S_BLIND_PNG)
     p = Predictor(model=model, dataset=ds)
     p.predict(upload=True, labels=Dataset.registry.LS10_TRAIN_224_PNG.labels)