search

lightly-ai / lightly

A python library for self-supervised learning on images.
https://docs.lightly.ai/self-supervised-learning/
MIT License
2.83k stars 246 forks source link

Return trainer from train_embedding #1571

Closed guarin closed 2 days ago

guarin commented 2 days ago

Changes

Returning the training helps checking the final training status. For example, trainer.interrupted can be checked with this to see whether the training completed or not.

codecov[bot] commented 2 days ago

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark:

Project coverage is 81.81%. Comparing base (bcf8ad1) to head (7ef052d).

Additional details and impacted files ```diff @@ Coverage Diff @@ ## master #1571 +/- ## ======================================= Coverage 81.81% 81.81% ======================================= Files 144 144 Lines 6119 6120 +1 ======================================= + Hits 5006 5007 +1 Misses 1113 1113 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.