File size: 564 Bytes
4c9e405
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
import torch

!pip install torch

from transformers import AutoModelForCausalLM, AutoTokenizer

model = AutoModelForCausalLM.from_pretrained("gpt2")
tokenizer = AutoTokenizer.from_pretrained("gpt2")

inputs = tokenizer("Hugging Face is a startup based in New York City and Paris",
                   return_tensors="pt")

loss = model(input_ids=inputs["input_ids"],
             labels=inputs["input_ids"]).loss

loss = model(input_ids=inputs["input_ids"],
             labels=inputs["input_ids"]).loss
ppl = torch.exp(loss)

print(f"Perplexity: {ppl.item(): 2f}")