jrzaurin / pytorch-widedeep

A flexible package for multimodal-deep-learning to combine tabular data with text and images using Wide and Deep models in Pytorch
Apache License 2.0
1.3k stars 190 forks source link

Option to save Optimizer in the `save` method #220

Closed jrzaurin closed 4 months ago

jrzaurin commented 4 months ago

Simple adjustment to add the option to save the optimizer when using the Trainer's save method

codecov[bot] commented 4 months ago

Codecov Report

Attention: Patch coverage is 58.16327% with 41 lines in your changes missing coverage. Please review.

Project coverage is 93.94%. Comparing base (8057360) to head (5b4bd9e).

Files Patch % Lines
...ed_training/_base_contrastive_denoising_trainer.py 48.64% 19 Missing :warning:
...pervised_training/_base_encoder_decoder_trainer.py 17.39% 19 Missing :warning:
pytorch_widedeep/training/_base_trainer.py 90.47% 2 Missing :warning:
pytorch_widedeep/training/trainer.py 83.33% 1 Missing :warning:
Additional details and impacted files ```diff @@ Coverage Diff @@ ## master #220 +/- ## ========================================== - Coverage 94.38% 93.94% -0.44% ========================================== Files 117 117 Lines 7445 7504 +59 ========================================== + Hits 7027 7050 +23 - Misses 418 454 +36 ```

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