fix bias size assert#5992
Conversation
|
@pytorchbot retest this please |
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
|
Also, can you please add a test for Bilinear without bias? Just to make sure we won't have a regression again |
|
@elanmart the onnx failure means it's asking you to rebase :) |
|
@pytorchbot retest this please. |
|
The onnx failure seems unrelated [1], but I can't force the tests to run again :( [1] |
|
@onnxbot retest this please |
| _assertGradAndGradgradChecks(self, lambda x1, x2: F.bilinear(x1, x2, module.weight, module.bias), | ||
| (input1_1, input2_1)) | ||
|
|
||
| def test_bilinear_no_bias(self): |
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
|
@pytorchbot retest this please |
| output_no_bias = module_no_bias(input1, input2) | ||
|
|
||
| self.assertEqual(output, output_no_bias) | ||
| output_no_bias.sum().backward() |
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
Fixes #5990