`yolo.py --profile` default GPU batch size 16 · 7830e91b9a - yolov5 - 拓恒技术版本管理系统

`yolo.py --profile` default GPU batch size 16

modifyDataloader

Glenn Jocher

GitHub 2 years ago

parent

3373aab56c

commit

7830e91b9a

No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23

1 changed files with 1 additions and 1 deletions

Split View

							
								+ 1
								
								- 1
							
models/yolo.pyView File
								
			@@ -314,7 +314,7 @@ if __name__ == '__main__':
		
			    # Profile
		
			    if opt.profile:
		
			        img = torch.rand(8 if torch.cuda.is_available() else 1, 3, 640, 640).to(device)
		
			        img = torch.rand(16 if torch.cuda.is_available() else 1, 3, 640, 640).to(device)
		
			        y = model(img, profile=True)
		
			    # Test all models