set model to eval when inferencing (very important)
This commit is contained in:
parent
b15c0fc33b
commit
4a799aa991
|
@ -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([
|
||||
|
|
Loading…
Reference in New Issue
Block a user