* track batch size after autobatch

* remove redundant import

* Update __init__.py

* Update __init__.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>