From 93f49d8987f3d92d1081334cc2eae17c3e45c6a4 Mon Sep 17 00:00:00 2001 From: Nic Ma Date: Wed, 26 Jan 2022 08:02:10 +0800 Subject: [PATCH] [DLMED] fix mode issue Signed-off-by: Nic Ma --- monai/engines/evaluator.py | 2 +- tests/test_prepare_batch_default.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/monai/engines/evaluator.py b/monai/engines/evaluator.py index b329462e24..e3e70d8b32 100644 --- 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 index 5417c8ca51..96051b5e82 100644 --- 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