We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1a44403 commit a187706Copy full SHA for a187706
brainpy/_src/dnn/tests/test_linear.py
@@ -70,7 +70,8 @@ def test_OneToOne(self):
70
]
71
)
72
def test_MaskedLinear(self, conn):
73
- f = bp.dnn.MaskedLinear(conn, weight=bp.init.XavierNormal())
+ bm.random.DEFAULT.seed(123)
74
+ f = bp.dnn.MaskedLinear(conn, weight=bp.init.XavierNormal(seed=123))
75
x = bm.random.random((16, 100))
76
y = f(x)
77
self.assertTrue(y.shape == (16, 100))
0 commit comments