Glenn Jocher
4 years ago
1 changed files with 1 additions and 1 deletions
+ 1
- 1
models/onnx_export.py
View File
| # 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() | ||||
Loading…