Multi-head model

[gitttt-1234]

This PR has all the modules required for multi-head training

[codecov]

Codecov Report

Attention: Patch coverage is 15.35484% with 656 lines in your changes missing coverage. Please review.

Project coverage is 85.76%. Comparing base (75f4368) to head (05c3fa8).
Report is 1 commits behind head on main.

Files with missing lines	Patch %	Lines
sleap_nn/training/lightning_modules.py	9.66%	346 Missing ⚠️
sleap_nn/training/model_trainer.py	4.30%	200 Missing ⚠️
sleap_nn/architectures/model.py	11.53%	46 Missing ⚠️
sleap_nn/inference/predictors.py	57.89%	40 Missing ⚠️
sleap_nn/training/utils.py	9.52%	19 Missing ⚠️
sleap_nn/inference/bottomup.py	71.42%	2 Missing ⚠️
sleap_nn/inference/topdown.py	60.00%	2 Missing ⚠️
sleap_nn/inference/single_instance.py	66.66%	1 Missing ⚠️

❌ Your patch check has failed because the patch coverage (15.35%) is below the target coverage (95.00%). You can increase the patch coverage or adjust the target coverage.

Additional details and impacted files

@@             Coverage Diff             @@
##             main     #188       +/-   ##
===========================================
- Coverage   96.68%   85.76%   -10.93%     
===========================================
  Files          47       47               
  Lines        5014     5773      +759     
===========================================
+ Hits         4848     4951      +103     
- Misses        166      822      +656     

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

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.