Added Sequence Classification class in GPTNeo (#11906)

* seq classification changes * fix tests
2021-05-28 15:57:02 +05:30
parent 80d712fac6
commit e1205e478a
9 changed files with 159 additions and 4 deletions
--- a/docs/source/model_doc/gpt_neo.rst
+++ b/docs/source/model_doc/gpt_neo.rst
@@ -65,3 +65,9 @@ GPTNeoForCausalLM

 .. autoclass:: transformers.GPTNeoForCausalLM
    :members: forward
+
+GPTNeoForSequenceClassification
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+.. autoclass:: transformers.GPTNeoForSequenceClassification
+    :members: forward