Model Trained Using AutoTrain

  • Problem type: Extractive Question Answering

Validation Metrics

loss: 5.707379341125488

exact_match: 0.0

f1: 0.0

runtime: 13.0624

samples_per_second: 0.766

steps_per_second: 0.077

: 3.0

Usage

import torch

from transformers import AutoModelForQuestionAnswering, AutoTokenizer

model = AutoModelForQuestionAnswering.from_pretrained(...)

tokenizer = AutoTokenizer.from_pretrained(...)

from transformers import BertTokenizer, BertForQuestionAnswering

question, text = "Who loves AutoTrain?", "Everyone loves AutoTrain"

inputs = tokenizer(question, text, return_tensors='pt')

start_positions = torch.tensor([1])

end_positions = torch.tensor([3])

outputs = model(**inputs, start_positions=start_positions, end_positions=end_positions)

loss = outputs.loss

start_scores = outputs.start_logits

end_scores = outputs.end_logits
Downloads last month
107
Safetensors
Model size
109M params
Tensor type
F32
·
Inference Providers NEW
This model is not currently available via any of the supported Inference Providers.

Model tree for Ai4des/autotrain-kjxi3-hql8x

Finetuned
(214)
this model