Commit 4811a3d1 authored by myhloli's avatar myhloli
Browse files

fix(pdf-extract): ensure model is set to evaluation mode before processing

Add model.eval() invocation to pdf_extract_kit initialization sequence to ensure the
model is set to evaluation mode. This is critical for proper inference and performance
metrics when processing PDF content.
parent c36fa049
...@@ -64,6 +64,7 @@ def mfr_model_init(weight_dir, cfg_path, _device_='cpu'): ...@@ -64,6 +64,7 @@ def mfr_model_init(weight_dir, cfg_path, _device_='cpu'):
task = tasks.setup_task(cfg) task = tasks.setup_task(cfg)
model = task.build_model(cfg) model = task.build_model(cfg)
model = model.to(_device_) model = model.to(_device_)
model.eval()
vis_processor = load_processor('formula_image_eval', cfg.config.datasets.formula_rec_eval.vis_processor.eval) vis_processor = load_processor('formula_image_eval', cfg.config.datasets.formula_rec_eval.vis_processor.eval)
mfr_transform = transforms.Compose([vis_processor, ]) mfr_transform = transforms.Compose([vis_processor, ])
return [model, mfr_transform] return [model, mfr_transform]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment