Feature/model persistence

[nsundalov]

Description

Allow save and load multiple times. Load train state on loading model.

Type of change

• Bug fix (non-breaking change which fixes an issue)
• New feature (non-breaking change which adds functionality)
• Breaking change (fix or feature that would cause existing functionality to not work as expected)
• Optimization

How Has This Been Tested?

Before submitting a PR, please check yourself against the following list. It would save us quite a lot of time.

• Have you read the contribution guide?
• Have you updated the relevant docstrings? We're using Numpy format, please double-check yourself
• Does your change require any new tests?
• Have you updated the changelog file?

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 100.00%. Comparing base (9b3992e) to head (8c50c4f).
⚠️ Report is 122 commits behind head on main.

Additional details and impacted files

@@             Coverage Diff             @@
##              main      #289     +/-   ##
===========================================
  Coverage   100.00%   100.00%             
===========================================
  Files           45        78     +33     
  Lines         2242      5346   +3104     
===========================================
+ Hits          2242      5346   +3104     

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

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.