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

Prepare classes for Poincare embedding #125

Open
wants to merge 14 commits into
base: develop
Choose a base branch
from
Open

Conversation

lhjohn
Copy link
Collaborator

@lhjohn lhjohn commented Aug 22, 2024

No description provided.

@egillax
Copy link
Collaborator

egillax commented Oct 8, 2024

@lhjohn I think you need to pull in latest develop to get the CI to run

* [WIP] custom embeddings

* [WIP]

* Finish refactor and fix test

* remove unnused dep

* fix action

* python311

* Add poincare class

* Fix mapping issue and add test

* register buffers for custom embeddings

* trainable embeddings

* make resnet and mlp responsible to ensure embeddings are correct dims

* add tests for poincare ResNet and MLP
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.

2 participants