Small edits to training

[sfmig]

Small edits to training to better match evaluation.

TODO:

• factor out logging to screen of MLflow and datasets info

[codecov-commenter]

Codecov Report

Attention: Patch coverage is 92.85714% with 1 line in your changes missing coverage. Please review.

Project coverage is 48.60%. Comparing base (7105c4c) to head (9b5ccc3).

Files with missing lines	Patch %	Lines
crabs/detector/train_model.py	92.85%	1 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #235      +/-   ##
==========================================
+ Coverage   47.69%   48.60%   +0.91%     
==========================================
  Files          24       24              
  Lines        1495     1506      +11     
==========================================
+ Hits          713      732      +19     
+ Misses        782      774       -8     

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