Skip to content

GitLab

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
test_modeling_tf_convnextv2.py 11.7 KB