Backbone add mixin tests (#22542)

* Add out_indices to backbones, deprecate out_features

* Update - can specify both out_features and out_indices but not both

* Add backbone mixin tests

* Test tidy up

* Add test_backbone for convnext

* Remove redefinition of method

* Update for Dinat and Nat backbones

* Update tests

* Smarter indexing

* Add checks on config creation for backbone

* PR comments
This commit is contained in:
amyeroberts
2023-04-06 13:50:15 +01:00
committed by GitHub
parent 48706c7178
commit 12d51db243
10 changed files with 316 additions and 2 deletions

View File

@@ -58,6 +58,7 @@ class ConvNextV2ModelTester:
num_labels=10,
initializer_range=0.02,
out_features=["stage2", "stage3", "stage4"],
out_indices=[2, 3, 4],
scope=None,
):
self.parent = parent
@@ -74,6 +75,7 @@ class ConvNextV2ModelTester:
self.num_labels = num_labels
self.initializer_range = initializer_range
self.out_features = out_features
self.out_indices = out_indices
self.scope = scope
def prepare_config_and_inputs(self):
@@ -97,6 +99,7 @@ class ConvNextV2ModelTester:
is_decoder=False,
initializer_range=self.initializer_range,
out_features=self.out_features,
out_indices=self.out_indices,
num_labels=self.num_labels,
)