[DLMED] fix mode issue (#3715)

Signed-off-by: Nic Ma <nma@nvidia.com>
Project-MONAI · Feb 9, 2022 · 647a100 · 647a100
1 parent abe24cf
commit 647a100
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/monai/engines/evaluator.py b/monai/engines/evaluator.py
@@ -114,7 +114,7 @@ def __init__(
             event_to_attr=event_to_attr,
             decollate=decollate,
         )
-        self.mode = look_up_option(mode, ForwardMode)
+        mode = look_up_option(mode, ForwardMode)
         if mode == ForwardMode.EVAL:
             self.mode = eval_mode
         elif mode == ForwardMode.TRAIN:

diff --git a/tests/test_prepare_batch_default.py b/tests/test_prepare_batch_default.py
@@ -43,6 +43,7 @@ def test_content(self):
             non_blocking=False,
             prepare_batch=PrepareBatchDefault(),
             decollate=False,
+            mode="eval",
         )
         evaluator.run()
         output = evaluator.state.output