SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

add FlaxAutoModelForImageClassification in main init (#12298)

Browse Source
This commit is contained in:
Suraj Patil
2021-06-22 18:26:05 +05:30
committed by GitHub
parent 2affeb2905
commit 1498eb9888
5 changed files with 27 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
docs/source/model_doc/auto.rst
View File

@@ -266,3 +266,10 @@ FlaxAutoModelForNextSentencePrediction
.. autoclass:: transformers.FlaxAutoModelForNextSentencePrediction
:members:
FlaxAutoModelForImageClassification
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.FlaxAutoModelForImageClassification
:members: