I have a paragraph for example below Either party may terminate this Agreement by written notice at any time if the other party defaults in the performance of its material obligations hereunder. In the event of such default, the party declaring the default shall provide the defaulting party with w…

Thanks, I understood, I am running your code on my colab when I run below code:- def train(epoch): tr_loss, tr_accuracy = 0, 0 nb_tr_examples, nb_tr_steps = 0, 0 tr_preds, tr_labels = [], [] # put model in training mode model.train() for idx, batch in enumerate(t…

@nielsr Can you take a look at above error?

Hey, could you share a reproducible notebook on Colab or Kaggle?

I am just following the notebook which is this:- Transformers-Tutorials/Custom_Named_Entity_Recognition_with_BERT.ipynb at master · NielsRogge/Transformers-Tutorials · GitHub and I only done some small changes which I listed above!

Try changing from: loss, tr_logits = model(input_ids=ids, attention_mask=mask, labels=targets) to: outputs = model(input_ids=ids, attention_mask=mask, labels=targets) loss = outputs["loss"] tr_logits = outputs["logits"]

[image] Emanuel: outputs = model(input_ids=ids, attention_mask=mask, labels=targets) loss = outputs["loss"] tr_logits = outputs["logits"] Hey, def train(epoch): tr_loss, tr_accuracy = 0, 0 nb_tr_examples, nb_tr_steps = 0, 0 tr_preds, tr_labels = [], [] # put model in trai…

@Emanuel You can see the collab which I am running:- Google Colab

Change from loss.items() to loss.item()

Thanks, It worked now I trained the model and I saved the model as well, so how do I load the model and make prediction, I am completely new to hugging face, so how do I load it and make prediction. I have saved the tokenizer and model. @Emanuel

How to fine tune bert on entity recognition?

Beginners

sbmaruf November 11, 2021, 2:13pm 10

May be this can help, Bert ner classifier

Topic		Replies	Views
Ask for help with prediction results of Named Entity Recognition Task 🤗Transformers	10	3337	May 21, 2021
Initialising BERT Model Models	0	316	June 9, 2023
How to deal with differences between CoNLL 2003 dataset tokenisation and BER tokeniser when fine tuning NER model? Intermediate	6	2886	November 23, 2021
BioBERT NER issue Beginners	7	4727	November 27, 2022
BERT for NER output of only '0' Beginners	0	703	November 14, 2021

How to fine tune bert on entity recognition?

Related topics