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

half() fix #66

This commit is contained in:
Glenn Jocher 2020-06-15 00:43:57 -07:00
parent 111994956e
commit e7c24e04ac
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
test.py
View File

@ -45,6 +45,7 @@ def test(data,
training = False training = False
else: # called by train.py else: # called by train.py
device = next(model.parameters()).device # get model device device = next(model.parameters()).device # get model device
half = False
training = True training = True
# Configure # Configure