Skip to content

Conversation

@sakares
Copy link
Contributor

@sakares sakares commented Mar 14, 2020

The current modeling_xlm.py did not have alike ForTokenClassification class like others, which helps for NER task comparison across all existing models.

now XLMForTokenClassification can be called via:

from transformers import XLMForTokenClassification

model = XLMForTokenClassification.from_pretrained('xlm-mlm-100-1280')

@sshleifer sshleifer changed the title Add the missing token classification for XLM Add missing token classification for XLM Mar 15, 2020
Copy link
Contributor

@sshleifer sshleifer left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@LysandreJik LysandreJik force-pushed the add-XLMForTokenClassification branch from a455a4b to 24c48b5 Compare March 26, 2020 13:40
Copy link
Member

@LysandreJik LysandreJik left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Cool, thanks @sakares! I rebased on your branch and added a commit for tests that should be slow (question answering and token classification in auto models)

@LysandreJik LysandreJik merged commit 1a6c546 into huggingface:master Mar 26, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants