model fuse · 546f973562 - yolov5 - 拓恒技术版本管理系统

model fuse

5.0

Glenn Jocher 4 years ago

parent

8fe299f179

commit

546f973562

1 changed files with 0 additions and 2 deletions

Unified View

							
								+ 0
								
								- 2
							
test.pyView File
								
															        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)