• 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
convnextv2.md 4.07 KB