AutoBatch protect from extreme batch sizes (#9209)
If < 1 or > 1024 set output to default batch size 16.
May partially address https://github.com/ultralytics/yolov5/issues/9156Signed-off-by:
Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by:
Glenn Jocher <glenn.jocher@ultralytics.com>
正在显示
请
注册
或者
登录
后发表评论