Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "76a1417f2af9bfad3ad7cd2a5d3ad5de9d5f5f13"
Switch branch/tag
Find file BlameHistoryPermalink
  • Andi Powers Holmes's avatar
    Add TensorFlow implementation of ConvNeXTv2 (#25558) · f8afb2b2
    Andi Powers Holmes authored 
    * 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.
    f8afb2b2
check_docstrings.py 39.3 KB