Glenn Jocher
4 years ago
1 changed files with 1 additions and 1 deletions
+ 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'].float().fuse() # load to FP32 | |||||
| model = torch.load(weights, map_location=device)['model'].float().fuse().to(device) # load to FP32 | |||||
| imgsz = check_img_size(imgsz, s=model.stride.max()) # check img_size | imgsz = check_img_size(imgsz, s=model.stride.max()) # check img_size | ||||
| # Multi-GPU disabled, incompatible with .half() https://github.com/ultralytics/yolov5/issues/99 | # Multi-GPU disabled, incompatible with .half() https://github.com/ultralytics/yolov5/issues/99 |
Loading…