Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
AutoBatch checks against failed solutions (ultralytics#8159)
* AutoBatch checks against failed solutions @kalenmike this is a simple improvement to AutoBatch to verify that returned solutions have not already failed, i.e. return batch-size 8 when 8 already produced CUDA out of memory. This is a halfway fix until I can implement a 'final solution' that will actively verify the solved-for batch size rather than passively assume it works. * Update autobatch.py * Update autobatch.py
- Loading branch information