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

Convert neuralop model checkpoints for version >0.2.0 #10

Open
arthurfeeney opened this issue Nov 4, 2023 · 1 comment
Open

Convert neuralop model checkpoints for version >0.2.0 #10

arthurfeeney opened this issue Nov 4, 2023 · 1 comment

Comments

@arthurfeeney
Copy link
Member

arthurfeeney commented Nov 4, 2023

It seems that neuralop refactored some of their internal data structures, so checkpoints that were trained with version 0.2.0 are not loading when using neuralop's main branch. This may only be the case when using the tuckerized version... 🤷 Not terrible, can just set 0.2.0 as the version w/ conda, but would be nice to work with later versions.

There may be a way to convert the checkpoints to the new version. Probably easiest to ask them in an issue instead.

@arthurfeeney
Copy link
Member Author

could also just condition the imports on the version of neuraloperator. Just require version 0.2.0 for reproducibility and running the old checkpoints, and try to use the most recent version for anything else.

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

No branches or pull requests

1 participant