From 4a799aa991e29b4134f26e0a58c289b7bf464f68 Mon Sep 17 00:00:00 2001 From: mrq Date: Sat, 5 Aug 2023 04:28:18 +0000 Subject: [PATCH] set model to eval when inferencing (very important) --- image_classifier/inference.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/image_classifier/inference.py b/image_classifier/inference.py index 9a74850..1e5e318 100755 --- a/image_classifier/inference.py +++ b/image_classifier/inference.py @@ -16,6 +16,8 @@ class CAPTCHA(): else: self.load_model_from_cfg( config ) + self.model.eval() + self.width = width self.height = height self.transform = transforms.Compose([