mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

ONNX export bug fix #93

Browse Source
This commit is contained in:
Glenn Jocher 2020-06-16 09:59:42 -07:00
parent 8b26e89006
commit e1e3399286
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models/onnx_export.py
View File

@ -25,7 +25,7 @@ if __name__ == '__main__':
# Load pytorch model # Load pytorch model
google_utils.attempt_download(opt.weights) google_utils.attempt_download(opt.weights)
model = torch.load(opt.weights, map_location=torch.device('cpu'))['model'] model = torch.load(opt.weights, map_location=torch.device('cpu'))['model'].float()
model.eval() model.eval()
model.fuse() model.fuse()