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

model fuse

This commit is contained in:
Glenn Jocher 2020-06-07 14:47:48 -07:00
parent 8fe299f179
commit 546f973562
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test.py
View File

@ -33,8 +33,6 @@ def test(data,
google_utils.attempt_download(weights) google_utils.attempt_download(weights)
model = torch.load(weights, map_location=device)['model'] model = torch.load(weights, map_location=device)['model']
torch_utils.model_info(model) torch_utils.model_info(model)
# Fuse
# model.fuse() # model.fuse()
model.to(device) model.to(device)