Skip to content

GitLab

Commit 80f968fe authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Make triangle attention test more sensitive

parent de91f011
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_triangular_attention.py
View file @ 80f968fe
...@@ -70,7 +70,7 @@ class TestTriangularAttention(unittest.TestCase): ...@@ -70,7 +70,7 @@ class TestTriangularAttention(unittest.TestCase):
n_res = consts.n_res n_res = consts.n_res
pair_act = np.random.rand(n_res, n_res, consts.c_z) pair_act = np.random.rand(n_res, n_res, consts.c_z) * 10
pair_mask = np.random.randint(low=0, high=2, size=(n_res, n_res)) pair_mask = np.random.randint(low=0, high=2, size=(n_res, n_res))
# Fetch pretrained parameters (but only from one block)] # Fetch pretrained parameters (but only from one block)]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment