set model to eval when inferencing (very important)

This commit is contained in:
mrq 2023-08-05 04:28:18 +00:00
parent b15c0fc33b
commit 4a799aa991

View File

@ -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([