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

Legacy model weights import #64

Open
talmo opened this issue Jul 3, 2024 · 0 comments
Open

Legacy model weights import #64

talmo opened this issue Jul 3, 2024 · 0 comments

Comments

@talmo
Copy link
Contributor

talmo commented Jul 3, 2024

Core SLEAP models are saved as Keras HDF5 files, which makes them fairly easily readable without actually depending on TensorFlow or Keras.

The task would be to understand the structure of those HDF5 files, load up the weights, possibly transpose the ordering to match the kernel ordering in PyTorch, and assign those values to the appropriate nn.modules.

See also: #63 which is a pre-requisite so that we can instantiate the models as PyTorch nn.modules in the first place.

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