Glenn Jocher
4 years ago
2 changed files with 2 additions and 2 deletions
+ 1
- 1
detect.py
View File
| # Load model | # Load model | ||||
| 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'].float() # load to FP32 | |||||
| # torch.save(torch.load(weights, map_location=device), weights) # update model if SourceChangeWarning | # torch.save(torch.load(weights, map_location=device), weights) # update model if SourceChangeWarning | ||||
| # model.fuse() | # model.fuse() | ||||
| model.to(device).eval() | model.to(device).eval() |
+ 1
- 1
test.py
View File
| # Load model | # Load model | ||||
| 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'].float() # load to FP32 | |||||
| torch_utils.model_info(model) | torch_utils.model_info(model) | ||||
| # model.fuse() | # model.fuse() | ||||
| model.to(device) | model.to(device) |
Loading…