Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Fix] Update model saving for TF backend #1782

Merged
merged 2 commits into from
Jun 20, 2024

Conversation

agniv-the-marker
Copy link
Contributor

TF save_weights

Currently the saving for backend saves to the wrong filetype, according to the docs it must end with a .weights.h5 suffix, currently ends in .ckpt suffix which breaks if you are using the TF backend.

Updated correct file endings for save files + cleaned up the code.
deepxde/model.py Outdated Show resolved Hide resolved
deepxde/model.py Outdated Show resolved Hide resolved
Revert with the correction.
@lululxvi lululxvi merged commit 51581b2 into lululxvi:master Jun 20, 2024
11 checks passed
g-w1 pushed a commit to g-w1/deepxde that referenced this pull request Jun 26, 2024
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.

None yet

2 participants