set model to eval when inferencing (very important)

2023-08-05 04:28:18 +00:00 · 2023-08-05 04:28:18 +00:00 · 4a799aa991
commit 4a799aa991
parent b15c0fc33b
1 changed files with 2 additions and 0 deletions
--- 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([