Change back to train mode after eval

Former-commit-id: 773ef215d41c1f36dc0ed4159c12df89d792fbc3
2020-05-05 09:15:07 -07:00 · 2020-05-05 09:15:07 -07:00 · b83c81f30b
parent 74a3f2496f
commit b83c81f30b
1 changed files with 1 additions and 0 deletions
--- a/eval.py
+++ b/eval.py
@ -29,4 +29,5 @@ def eval_net(net, loader, device):
                tot += dice_coeff(pred, true_masks).item()
            pbar.update()

+    net.train()
    return tot / n_val