-
Notifications
You must be signed in to change notification settings - Fork 71
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
Fix bug in Collector with Graph data #456
Conversation
f32d7e4
to
6f52c79
Compare
6c66cf6
to
f32af3c
Compare
pina/utils.py
Outdated
@@ -48,7 +48,8 @@ def labelize_forward(forward, input_variables, output_variables): | |||
:type output_variables: list[str] | tuple[str] | |||
""" | |||
def wrapper(x): | |||
x = x.extract(input_variables) | |||
if isinstance(x, LabelTensor): |
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.
why this? This method should not check the instance since it is applied only when use_lt
is true in the solver. Please remove it.
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.
If we use graphs x is a Batch object. It is necessary if I want to use graphs and LabelTensors together
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.
I still don't see the point. In the solver do you set use_lt
to true?
pina/utils.py
Outdated
@@ -48,7 +48,8 @@ def labelize_forward(forward, input_variables, output_variables): | |||
:type output_variables: list[str] | tuple[str] | |||
""" | |||
def wrapper(x): | |||
x = x.extract(input_variables) | |||
if isinstance(x, LabelTensor): |
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.
I still don't see the point. In the solver do you set use_lt
to true?
f32af3c
to
78bedd0
Compare
78bedd0
to
a919d06
Compare
a919d06
to
37d61ea
Compare
This PR fix some bugs with solvers and Graphs