apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix updated image size not used (#739)

Browse Source 
Co-authored-by: magangloff <71378343+magangloff@users.noreply.github.com>
This commit is contained in:
Maximilian Gangloff 2022-09-12 16:45:16 +02:00 committed by GitHub
parent 2e99efb27f
commit b1850c7dca
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test.py
View File

@ -59,7 +59,7 @@ def test(data,
imgsz = check_img_size(imgsz, s=gs) # check img_size imgsz = check_img_size(imgsz, s=gs) # check img_size
if trace: if trace:
model = TracedModel(model, device, opt.img_size) model = TracedModel(model, device, imgsz)
# Half # Half
half = device.type != 'cpu' and half_precision # half precision only supported on CUDA half = device.type != 'cpu' and half_precision # half precision only supported on CUDA