Skip to content

Add save_model helper for ML-dSGP4 - #30

Open
sylvesterkaczmarek wants to merge 2 commits into
esa:masterfrom
sylvesterkaczmarek:feat/mldsgp4-save-model
Open

Add save_model helper for ML-dSGP4#30
sylvesterkaczmarek wants to merge 2 commits into
esa:masterfrom
sylvesterkaczmarek:feat/mldsgp4-save-model

Conversation

@sylvesterkaczmarek

Copy link
Copy Markdown

Summary

This adds a small persistence helper to mldsgp4:

  • mldsgp4.save_model(path), matching the existing load_model() API
  • a round-trip unit test that saves model weights, reloads them into a fresh model, verifies exact state_dict equality, and confirms the restored model is in evaluation mode

Context

This is a deliberately narrower follow-up to #29.

The feedback on #29 was that the synthetic target used in the proposed training notebook did not represent ML-dSGP4's intended use case of learning against high-precision dynamics. I agree with that concern, so this PR does not reintroduce the training tutorial or synthetic-data example.

The save/load functionality is independent of the choice of training data and directly addresses the model-persistence part of #19. The end-to-end training tutorial can then be handled separately once suitable real ephemeris data is available.

Scope

No propagation or training behaviour is changed. This PR only adds the save helper and its regression test.

Related: #19, #29

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant