mmazuecos
commited on
Commit
·
a6155e4
1
Parent(s):
22ee8aa
Updated config with model_type.
Browse files- .gitattributes +1 -0
- config.json +1 -0
.gitattributes
CHANGED
@@ -30,3 +30,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
30 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
31 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
32 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
30 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
31 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
32 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
33 |
+
pytorch_model.bin filter=lfs diff=lfs merge=lfs -text
|
config.json
CHANGED
@@ -6,6 +6,7 @@
|
|
6 |
"dropout_p": 0.1,
|
7 |
"embedding_size": 256,
|
8 |
"mask_pooling": false,
|
|
|
9 |
"num_conv": 8,
|
10 |
"num_heads": 8,
|
11 |
"num_layers": 6,
|
|
|
6 |
"dropout_p": 0.1,
|
7 |
"embedding_size": 256,
|
8 |
"mask_pooling": false,
|
9 |
+
"model_type": "rec_transformer",
|
10 |
"num_conv": 8,
|
11 |
"num_heads": 8,
|
12 |
"num_layers": 6,
|