guymorganb
commited on
Commit
·
b3294bc
1
Parent(s):
fa7462c
messing around with differing settings.
Browse files- modeling_lsg_bert.py +2 -2
modeling_lsg_bert.py
CHANGED
@@ -62,13 +62,13 @@ class LSGBertConfig(BertConfig):
|
|
62 |
base_model_prefix="lsg",
|
63 |
block_size=128,
|
64 |
sparse_block_size=128,
|
65 |
-
sparsity_factor=2,
|
66 |
adaptive=True,
|
|
|
67 |
lsh_num_pre_rounds=1,
|
68 |
mask_first_token=False,
|
69 |
num_global_tokens=1,
|
70 |
pool_with_global=True,
|
71 |
-
sparsity_type="
|
72 |
**kwargs
|
73 |
):
|
74 |
"""Constructs LSGBertConfig."""
|
|
|
62 |
base_model_prefix="lsg",
|
63 |
block_size=128,
|
64 |
sparse_block_size=128,
|
|
|
65 |
adaptive=True,
|
66 |
+
sparsity_factor=2,
|
67 |
lsh_num_pre_rounds=1,
|
68 |
mask_first_token=False,
|
69 |
num_global_tokens=1,
|
70 |
pool_with_global=True,
|
71 |
+
sparsity_type="none",
|
72 |
**kwargs
|
73 |
):
|
74 |
"""Constructs LSGBertConfig."""
|