From 47bdc77bce27907c19144023d496a9009685f075 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 11 Dec 2019 06:31:04 -0800 Subject: [PATCH] remove summary --- tests/utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/utils.py b/tests/utils.py index cdd32293..16a6a19f 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -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