From b1ac7151b4a73f8c119b7d6a0600faa7caa56a80 Mon Sep 17 00:00:00 2001 From: milesial Date: Wed, 26 Sep 2018 08:57:10 +0200 Subject: [PATCH] Switch net to eval mode in eval.py Former-commit-id: 3ab827241455818c7b81172a426d805f9ace0f8f --- eval.py | 1 + 1 file changed, 1 insertion(+) diff --git a/eval.py b/eval.py index 944c111..57c9972 100644 --- a/eval.py +++ b/eval.py @@ -6,6 +6,7 @@ from dice_loss import dice_coeff def eval_net(net, dataset, gpu=False): """Evaluation without the densecrf with the dice coefficient""" + net.eval() tot = 0 for i, b in enumerate(dataset): img = b[0]