remove summary

2026-09-10 12:21:57 +08:00 · 2019-12-11 06:31:04 -08:00
parent 6c5a6a1b4d
commit 47bdc77bce
1 changed files with 2 additions and 0 deletions
@@ -144,6 +144,8 @@ def load_model(exp, root_weights_dir, module_class=LightningTemplateModel):


 def run_prediction(dataloader, trained_model, dp=False, min_acc=0.50):
+    trained_model.eval()
+
    # run prediction on 1 batch
    for batch in dataloader:
        break