-
-
Notifications
You must be signed in to change notification settings - Fork 77
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
Product manifold #109
Product manifold #109
Conversation
@rrkarim you are welcome to review the PR. If anything is not clear enough, I will elaborate on docs more |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
You do not register bound methods in scaled manifold for methods like geodesic
. So something like this shouldn't work:
ball = geoopt.PoincareBallExact()
sball = geoopt.Scaled(ball, 2)
rsball = geoopt.Scaled(sball, 0.5)
rsball.geodesic
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Haven't checked tests yet, can you please fix please fix travis
fails and I will check the tests tomorrow, other than that I guess we can finally merge it.
@rrkarim I've fixed tests, they were related to black updates |
Thanks! |
* draft product manifold class * add initial progress on some methods * test slicing * add projections * draft other methods * add tests and fix broadcasting bugs * add typehints and make sure there are no more broadcasting problems * fix lint issue * fix lint issues * fix dist * add scaling utils * fix typo * fix inner product result * add scalings for poincare * scaling rescaling test * fix dir method * empty tensor fix * add a bit more tests for product manifold * fix typo * some nitpicka * add random methods and device/dtype consistency when __init__ * rest random * remove annotation * fix lint errors * dist test * make scaled implementation much cleaner * add comments * fix docstring * add restricted methods * black * better name * fix logmap * fix logmap * fix logmap * add component inner method * change doc * fix doc * add origin * update changelog * try o avoid warnings in multigpu env with origin * add more docs for Scaling Info * add docs * fix doc * method directive is not valif * origin docs * fix lineskip * update readme * fix attributes * improve component inner for euclidean manifold * fix broadcasting * fix typo * fix signature * black * update black
Adding Product Manifold there