midas/inspec
Viewer • Updated • 6k • 600 • 12
How to use artemfilipenko/keyphrase-generation-bart-large-trained-on-augmented-and-default-inspec with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("artemfilipenko/keyphrase-generation-bart-large-trained-on-augmented-and-default-inspec")
model = AutoModelForSeq2SeqLM.from_pretrained("artemfilipenko/keyphrase-generation-bart-large-trained-on-augmented-and-default-inspec", device_map="auto")This model is a fine-tuned version of facebook/bart-large on the midas/inspec generation dataset, concatenated with data augmented custom artemfilipenko/synonyms-augmented-5x-inspec dataset. It achieves the following results on the evaluation set:
More information needed
More information needed
More information needed
The following hyperparameters were used during training: