how to replace with other masked language model

marcotcr / checklist

Beyond Accuracy: Behavioral Testing of NLP models with CheckList

MIT License

2.01k stars 204 forks source link

how to replace with other masked language model #60

Closed feng-1985 closed 3 years ago

marcotcr commented 3 years ago

When you initialize Editor, use the argument model_name (default is roberta-base). The backend will initialize a tokenizer and a model using AutoTokenizer.from_pretrained(model_name) and AutoModelForMaskedLM.from_pretrained(model_name), so either a name or a path should work.

feng-1985 commented 3 years ago

Thanks. It seems should use the model listed on https://huggingface.co/models.