Skip to content

Instantly share code, notes, and snippets.

@remi-or

remi-or/roberta_peek.py

Created January 17, 2022 18:04
Show Gist options
  • Save remi-or/453b3a5646150ec9c6a4229e28497390 to your computer and use it in GitHub Desktop.
Save remi-or/453b3a5646150ec9c6a4229e28497390 to your computer and use it in GitHub Desktop.
Download ZIP
Roberta peek
Raw
roberta_peek.py
from transformers import AutoModelForMaskedLM
roberta = AutoModelForMaskedLM.from_pretrained("roberta-large")
print(roberta)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment