arxyzan commited on
Commit
f518871
1 Parent(s): 1a9f82a

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
2
  config_type: model
3
+ task: backbone
4
  activation: gelu
5
  attention_dropout: 0.1
6
  dim: 768