mrq/resnet-classifier
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

set model to eval when inferencing (very important)

Browse Source
This commit is contained in:
mrq 2023-08-05 04:28:18 +00:00
parent b15c0fc33b
commit 4a799aa991
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
image_classifier/inference.py
View File

@ -16,6 +16,8 @@ class CAPTCHA():
else: else:
self.load_model_from_cfg( config ) self.load_model_from_cfg( config )
self.model.eval()
self.width = width self.width = width
self.height = height self.height = height
self.transform = transforms.Compose([ self.transform = transforms.Compose([