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

Add TensorFlow implementation of ConvNeXTv2 (#25558)

Browse Source 
* Add type annotations to TFConvNextDropPath

* Use tf.debugging.assert_equal for TFConvNextEmbeddings shape check

* Add TensorFlow implementation of ConvNeXTV2

* check_docstrings: add TFConvNextV2Model to exclusions

TFConvNextV2Model and TFConvNextV2ForImageClassification have docstrings
which are equivalent to their PyTorch cousins, but a parsing issue prevents them
from passing the test.

Adding exclusions for these two classes as discussed in #25558.
This commit is contained in:
Andi Powers Holmes
2023-11-02 02:09:55 +11:00
committed by GitHub
parent 391d14e810
commit f8afb2b2ec
12 changed files with 1012 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
utils/check_docstrings.py
View File

@@ -533,6 +533,8 @@ OBJECTS_TO_IGNORE = [
"TFConvBertModel",
"TFConvNextForImageClassification",
"TFConvNextModel",
"TFConvNextV2Model", # Parsing issue. Equivalent to PT ConvNextV2Model, see PR #25558
"TFConvNextV2ForImageClassification",
"TFCvtForImageClassification",
"TFCvtModel",
"TFDPRReader",