mmazuecos commited on
Commit
a6155e4
·
1 Parent(s): 22ee8aa

Updated config with model_type.

Browse files
Files changed (2) hide show
  1. .gitattributes +1 -0
  2. 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,