Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

model fuse

Browse Source
This commit is contained in:
Glenn Jocher committed 2020-06-07 14:47:48 -07:00
1 parent 8fe299f179
commit 546f973562
1 file changed
-2
Download Patch File Download Diff File Expand all files Collapse all files
test.py
-2
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)