Browse Source

find a bug that bilinear must have bias

tags/v0.1.0
xuyige 6 years ago
parent
commit
246908bf45
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      test/modules/test_other_modules.py

+ 1
- 1
test/modules/test_other_modules.py View File

@@ -27,4 +27,4 @@ class TestBiLinear(unittest.TestCase):
x_right = torch.randn((7, 10, 20, 5)) x_right = torch.randn((7, 10, 20, 5))
y = bl(x_left, x_right) y = bl(x_left, x_right)
print(bl) print(bl)
bl2 = BiLinear(n_left=15, n_right=15, n_out=10, bias=False)
bl2 = BiLinear(n_left=15, n_right=15, n_out=10, bias=True)

Loading…
Cancel
Save