“Exemple d'albert pré-entraîné” Réponses codées

Exemple d'albert pré-entraîné


from transformers import AlbertTokenizer, AlbertForQuestionAnswering
import torch

tokenizer = AlbertTokenizer.from_pretrained('albert-base-v2')
model = AlbertForQuestionAnswering.from_pretrained('albert-base-v2')
question, text = "Who was Jim Henson?", "Jim Henson was a nice puppet"
input_dict = tokenizer.encode_plus(question, text, return_tensors='pt')
start_scores, end_scores = model(**input_dict)
Successful Shrike

Exemple d'albert pré-entraîné


from transformers import AlbertTokenizer, AlbertForQuestionAnswering
import torch

MODEL_PATH = 'ktrapeznikov/albert-xlarge-v2-squad-v2';

tokenizer = AlbertTokenizer.from_pretrained(MODEL_PATH)
model = AlbertForQuestionAnswering.from_pretrained(MODEL_PATH)

Modern Mongoose

Réponses similaires à “Exemple d'albert pré-entraîné”

Questions similaires à “Exemple d'albert pré-entraîné”

Parcourir les réponses de code populaires par langue

Parcourir d'autres langages de code