From fa68ffa78732b5950fb0dae72c115423ca2dd597 Mon Sep 17 00:00:00 2001 From: Weilin Xu Date: Mon, 8 Jun 2026 15:48:18 -0700 Subject: [PATCH] Run PyTorch models in eval mode by default. Signed-off-by: Weilin Xu --- src/xe_forge/core/executor.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/xe_forge/core/executor.py b/src/xe_forge/core/executor.py index 8debed4..864e93e 100644 --- a/src/xe_forge/core/executor.py +++ b/src/xe_forge/core/executor.py @@ -269,6 +269,8 @@ def _check_correctness( return False if original_model is not None: + # Run in eval mode instead of training mode. + original_model.eval() original_model = original_model.to(device) if dtype is not None: original_model = original_model.to(dtype) @@ -289,6 +291,8 @@ def _check_correctness( return False if optimized_model is not None: + # Run in eval mode instead of training mode. + optimized_model.eval() optimized_model = optimized_model.to(device) if dtype is not None: optimized_model = optimized_model.to(dtype)