Instructions to use hosung1/roberta_small_mlm_from_scratch with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use hosung1/roberta_small_mlm_from_scratch with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="hosung1/roberta_small_mlm_from_scratch")# Load model directly from transformers import AutoTokenizer, AutoModelForMaskedLM tokenizer = AutoTokenizer.from_pretrained("hosung1/roberta_small_mlm_from_scratch") model = AutoModelForMaskedLM.from_pretrained("hosung1/roberta_small_mlm_from_scratch") - Notebooks
- Google Colab
- Kaggle
roberta_small_mlm_from_scratch
RoBERTa-small을 Codenet데이터로 pre_train한 가중치입니다.
How to use
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tok = AutoTokenizer.from_pretrained("hosung1/roberta_small_mlm_from_scratch")
mdl = AutoModelForSequenceClassification.from_pretrained("hosung1/roberta_small_mlm_from_scratch")
- Downloads last month
- 6