Join GitHub today
GitHub is home to over 50 million developers working together to host and review code, manage projects, and build software together.
Sign upGitHub is where the world builds software
Millions of developers and companies build, ship, and maintain their software on GitHub — the largest and most advanced development platform in the world.
Implement a TF2 version of `GPT2ForSequenceClassification` #7622
Comments
May I take a crack at this? :) |
Yes, you may! Thanks @y2s82! |
Hi @LysandreJik , I have completed development for this FR. can you please assign it to me, so i'll raise PR for it. |
Feel free to open a PR |
The architecture
GPT2ForSequenceClassification
was added in #7501 in PyTorch. It would be great to have it in TensorFlow as well.Below is a list of items to follow to make sure the integration is complete:
TFGPT2ForSequenceClassification
inmodeling_tf_gpt2.py
tests/test_modeling_tf_gpt2.py
__init__.py
anddocs/source/model_doc/gpt2.rst
.Taking a look at the code changes in #7501 would be a good start, as this PR would essentially be a TF2 copy of it.
A very good first issue to get acquainted with the library and its architectures!