CAREamics / careamics

A deep-learning library for N2V and friends
https://careamics.github.io/
BSD 3-Clause "New" or "Revised" License
28 stars 5 forks source link

Implemented `training_step` and `validation_step`, fixed all the configs needed to instantiate the lightning module #218

Closed federico-carrara closed 1 month ago

federico-carrara commented 1 month ago

Description

This PR finally implements a working training loop for the lightining module VAEModule.

Changes Made

Things that can be improved


Please ensure your PR meets the following requirements:

codecov[bot] commented 1 month ago

Codecov Report

Attention: Patch coverage is 94.39252% with 6 lines in your changes missing coverage. Please review.

Project coverage is 87.20%. Comparing base (692429e) to head (35c2901). Report is 1 commits behind head on main.

Files Patch % Lines
src/careamics/lightning/lightning_module.py 93.33% 3 Missing :warning:
src/careamics/models/lvae/noise_models.py 62.50% 3 Missing :warning:
Additional details and impacted files ```diff @@ Coverage Diff @@ ## main #218 +/- ## ========================================== + Coverage 85.80% 87.20% +1.39% ========================================== Files 130 130 Lines 4799 4876 +77 ========================================== + Hits 4118 4252 +134 + Misses 681 624 -57 ```

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