search

LuxDL / Lux.jl

Elegant & Performant Scientific Machine Learning in Julia
https://lux.csail.mit.edu/
MIT License
446 stars 50 forks source link

Quality of Life Improvements #666

Closed avik-pal closed 1 month ago

avik-pal commented 1 month ago
  1. Fixes #664
  2. Takes the improvements from https://github.com/LuxDL/Lux.jl/pull/661
  3. Optimisers.adjust! and Optimisers.adjust can be directly applied to TrainState.
  4. StatefulLuxLayer has pretty printing
  5. StatefulLuxLayer is compatible with Adapt, so gpu_device() / cpu_device() can be directly applied to them.
  6. SimpleChains model printing now displays the internal model
codecov[bot] commented 1 month ago

Codecov Report

Attention: Patch coverage is 78.72340% with 10 lines in your changes are missing coverage. Please review.

Project coverage is 86.60%. Comparing base (ca2c635) to head (b3da5d3). Report is 2 commits behind head on main.

Files Patch % Lines
src/layers/display.jl 50.00% 5 Missing :warning:
src/helpers/stateful.jl 81.25% 3 Missing :warning:
src/layers/extension.jl 50.00% 2 Missing :warning:
Additional details and impacted files ```diff @@ Coverage Diff @@ ## main #666 +/- ## ========================================== - Coverage 86.87% 86.60% -0.27% ========================================== Files 50 50 Lines 2491 2524 +33 ========================================== + Hits 2164 2186 +22 - Misses 327 338 +11 ```

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