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

Fix a GPU memory leak in detect. No need to calculate gradients in inference. (#900)

Browse Source 
Co-authored-by: JKO095 <juho-pekka.koponen@wartsila.com>
This commit is contained in:
jpkoponen 2022-10-08 00:50:04 +03:00 committed by GitHub
parent 8b616af63a
commit 072f76c72c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
detect.py
View File

@ -84,6 +84,7 @@ def detect(save_img=False):
# Inference # Inference
t1 = time_synchronized() t1 = time_synchronized()
with torch.no_grad(): # Calculating gradients would cause a GPU memory leak
pred = model(img, augment=opt.augment)[0] pred = model(img, augment=opt.augment)[0]
t2 = time_synchronized() t2 = time_synchronized()