arxyzan commited on
Commit
cc51b6a
1 Parent(s): 384ccf7

Update model_config.yaml

Browse files
Files changed (1) hide show
  1. model_config.yaml +2 -2
model_config.yaml CHANGED
@@ -1,6 +1,6 @@
1
- name: distilbert_lm
2
  config_type: model
3
- task: language_modeling
4
  activation: gelu
5
  attention_dropout: 0.1
6
  dim: 768
 
1
+ name: distilbert_mask_filling
2
  config_type: model
3
+ task: mask_filling
4
  activation: gelu
5
  attention_dropout: 0.1
6
  dim: 768