Training loop stuff

[supersergiy]

No description provided.

[codecov]

Codecov Report

Attention: Patch coverage is 43.75000% with 27 lines in your changes missing coverage. Please review.

Project coverage is 99.55%. Comparing base (0dca420) to head (cd587e7).
Report is 1 commits behind head on main.

Files with missing lines	Patch %	Lines
...etta_utils/training/datasets/collection_dataset.py	44.73%	21 Missing ⚠️
zetta_utils/geometry/bbox.py	33.33%	6 Missing ⚠️

Additional details and impacted files

@@             Coverage Diff             @@
##              main     #821      +/-   ##
===========================================
- Coverage   100.00%   99.55%   -0.46%     
===========================================
  Files          142      143       +1     
  Lines         5952     6000      +48     
===========================================
+ Hits          5952     5973      +21     
- Misses           0       27      +27     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.