zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Explicit `model.eval()` call `if opt.train=False` (#3475) 

* call model.eval() when opt.train is False

call model.eval() when opt.train is False

* single-line if statement

* cleanup

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Yonghye Kwon 2021-06-07 01:39:36 +09:00 committed by GitHub
parent cf4f95bc5f
commit a1c3572bc9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
models/export.py
View File

@ -58,8 +58,7 @@ if __name__ == '__main__':
# Update model # Update model
if opt.half: if opt.half:
img, model = img.half(), model.half() # to FP16 img, model = img.half(), model.half() # to FP16
if opt.train: model.train() if opt.train else model.eval() # training mode = no Detect() layer grid construction
model.train() # training mode (no grid construction in Detect layer)
for k, m in model.named_modules(): for k, m in model.named_modules():
m._non_persistent_buffers_set = set() # pytorch 1.6.0 compatibility m._non_persistent_buffers_set = set() # pytorch 1.6.0 compatibility
if isinstance(m, models.common.Conv): # assign export-friendly activations if isinstance(m, models.common.Conv): # assign export-friendly activations