arxyzan commited on
Commit
1d5c815
1 Parent(s): 8f967bf

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: roberta_lm
2
  config_type: model
3
- task: language_modeling
4
  attention_probs_dropout_prob: 0.1
5
  bos_token_id: 0
6
  eos_token_id: 2
 
1
+ name: roberta_mask_filling
2
  config_type: model
3
+ task: mask_filling
4
  attention_probs_dropout_prob: 0.1
5
  bos_token_id: 0
6
  eos_token_id: 2