-
Notifications
You must be signed in to change notification settings - Fork 50
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
feature[design]: allow design plugin to interface with pytorch via Result #2189
Conversation
converting now to a PR since the notebook PR is out |
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.
Thanks @groberts-flex the accessors definitely are a great addition!
ce8bc51
to
9a166b0
Compare
Thanks @groberts-flex I think this is good to go implementation-wise, but still needs tests and a changelog entry. For the tests, I think the accessor methods can be tested easily (check |
86a8d2b
to
eaa243f
Compare
tests and changelog entry are in, how does this look for merging @yaugenst-flex? |
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.
Thanks @groberts-flex I think this looks good, only a minor comment.
0fb2b3f
to
f0af5b7
Compare
f0af5b7
to
06c33fc
Compare
This PR fixes a small bug in the design plugin wherein the optimization method returns the dimensions in a different order and so labeling with the
self.dims
parameter can end up being incorrect.It also implements accessor and length functions for the Result class so that it can be used directly as a Dataset in PyTorch. This also may be useful generally.
Marking this as a draft for now to discuss if we want to make these changes to the plugin. The changes are currently needed for the surrogate notebooks draft PR. (flexcompute/tidy3d-notebooks#228)