Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
609e6622
Unverified
Commit
609e6622
authored
Jun 19, 2024
by
amyeroberts
Committed by
GitHub
Jun 19, 2024
Browse files
Use self.config_tester.run_common_tests() (#31431)
* First testing updating config tests * Use run_common_tests
parent
7c71b61d
Changes
28
Hide whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
116 additions
and
128 deletions
+116
-128
src/transformers/models/align/configuration_align.py
src/transformers/models/align/configuration_align.py
+1
-1
src/transformers/models/timm_backbone/configuration_timm_backbone.py
...rmers/models/timm_backbone/configuration_timm_backbone.py
+1
-1
tests/models/align/test_modeling_align.py
tests/models/align/test_modeling_align.py
+6
-7
tests/models/bit/test_modeling_bit.py
tests/models/bit/test_modeling_bit.py
+4
-7
tests/models/convnext/test_modeling_convnext.py
tests/models/convnext/test_modeling_convnext.py
+8
-7
tests/models/convnextv2/test_modeling_convnextv2.py
tests/models/convnextv2/test_modeling_convnextv2.py
+8
-7
tests/models/cpmant/test_modeling_cpmant.py
tests/models/cpmant/test_modeling_cpmant.py
+1
-6
tests/models/cvt/test_modeling_cvt.py
tests/models/cvt/test_modeling_cvt.py
+8
-7
tests/models/deformable_detr/test_modeling_deformable_detr.py
...s/models/deformable_detr/test_modeling_deformable_detr.py
+7
-7
tests/models/depth_anything/test_modeling_depth_anything.py
tests/models/depth_anything/test_modeling_depth_anything.py
+6
-8
tests/models/dinat/test_modeling_dinat.py
tests/models/dinat/test_modeling_dinat.py
+4
-7
tests/models/donut/test_modeling_donut_swin.py
tests/models/donut/test_modeling_donut_swin.py
+8
-7
tests/models/efficientnet/test_modeling_efficientnet.py
tests/models/efficientnet/test_modeling_efficientnet.py
+6
-7
tests/models/focalnet/test_modeling_focalnet.py
tests/models/focalnet/test_modeling_focalnet.py
+8
-7
tests/models/grounding_dino/test_modeling_grounding_dino.py
tests/models/grounding_dino/test_modeling_grounding_dino.py
+7
-7
tests/models/levit/test_modeling_levit.py
tests/models/levit/test_modeling_levit.py
+4
-7
tests/models/maskformer/test_modeling_maskformer_swin.py
tests/models/maskformer/test_modeling_maskformer_swin.py
+8
-7
tests/models/perceiver/test_modeling_perceiver.py
tests/models/perceiver/test_modeling_perceiver.py
+7
-7
tests/models/regnet/test_modeling_regnet.py
tests/models/regnet/test_modeling_regnet.py
+7
-7
tests/models/resnet/test_modeling_resnet.py
tests/models/resnet/test_modeling_resnet.py
+7
-7
No files found.
src/transformers/models/align/configuration_align.py
View file @
609e6622
...
...
@@ -193,7 +193,7 @@ class AlignVisionConfig(PretrainedConfig):
hidden_act (`str` or `function`, *optional*, defaults to `"silu"`):
The non-linear activation function (function or string) in each block. If string, `"gelu"`, `"relu"`,
`"selu", `"gelu_new"`, `"silu"` and `"mish"` are supported.
hidde
m
_dim (`int`, *optional*, defaults to 1280):
hidde
n
_dim (`int`, *optional*, defaults to 1280):
The hidden dimension of the layer before the classification head.
pooling_type (`str` or `function`, *optional*, defaults to `"mean"`):
Type of final pooling to be applied before the dense classification head. Available options are [`"mean"`,
...
...
src/transformers/models/timm_backbone/configuration_timm_backbone.py
View file @
609e6622
...
...
@@ -79,5 +79,5 @@ class TimmBackboneConfig(PretrainedConfig):
self
.
features_only
=
features_only
self
.
use_pretrained_backbone
=
use_pretrained_backbone
self
.
use_timm_backbone
=
True
self
.
out_indices
=
out_indices
if
out_indices
is
not
None
else
(
-
1
,)
self
.
out_indices
=
out_indices
if
out_indices
is
not
None
else
[
-
1
]
self
.
freeze_batch_norm_2d
=
freeze_batch_norm_2d
tests/models/align/test_modeling_align.py
View file @
609e6622
...
...
@@ -147,16 +147,15 @@ class AlignVisionModelTest(ModelTesterMixin, unittest.TestCase):
def
setUp
(
self
):
self
.
model_tester
=
AlignVisionModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
AlignVisionConfig
,
has_text_modality
=
False
,
hidden_size
=
37
self
,
config_class
=
AlignVisionConfig
,
has_text_modality
=
False
,
hidden_size
=
37
,
common_properties
=
[
"num_channels"
,
"image_size"
],
)
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
self
.
config_tester
.
run_common_tests
()
@
unittest
.
skip
(
reason
=
"AlignVisionModel does not use inputs_embeds"
)
def
test_inputs_embeds
(
self
):
...
...
tests/models/bit/test_modeling_bit.py
View file @
609e6622
...
...
@@ -173,15 +173,12 @@ class BitModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
def
setUp
(
self
):
self
.
model_tester
=
BitModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
BitConfig
,
has_text_modality
=
False
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
BitConfig
,
has_text_modality
=
False
,
common_properties
=
[
"num_channels"
]
)
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
self
.
config_tester
.
run_common_tests
()
@
unittest
.
skip
(
reason
=
"Bit does not output attentions"
)
def
test_attention_outputs
(
self
):
...
...
tests/models/convnext/test_modeling_convnext.py
View file @
609e6622
...
...
@@ -183,15 +183,16 @@ class ConvNextModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase
def
setUp
(
self
):
self
.
model_tester
=
ConvNextModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
ConvNextConfig
,
has_text_modality
=
False
,
hidden_size
=
37
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
ConvNextConfig
,
has_text_modality
=
False
,
hidden_size
=
37
,
common_properties
=
[
"num_channels"
,
"hidden_sizes"
],
)
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
self
.
config_tester
.
run_common_tests
()
@
unittest
.
skip
(
reason
=
"ConvNext does not use inputs_embeds"
)
def
test_inputs_embeds
(
self
):
...
...
tests/models/convnextv2/test_modeling_convnextv2.py
View file @
609e6622
...
...
@@ -191,15 +191,16 @@ class ConvNextV2ModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCa
def
setUp
(
self
):
self
.
model_tester
=
ConvNextV2ModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
ConvNextV2Config
,
has_text_modality
=
False
,
hidden_size
=
37
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
ConvNextV2Config
,
has_text_modality
=
False
,
hidden_size
=
37
,
common_properties
=
[
"hidden_sizes"
,
"num_channels"
],
)
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
self
.
config_tester
.
run_common_tests
()
@
unittest
.
skip
(
reason
=
"ConvNextV2 does not use inputs_embeds"
)
def
test_inputs_embeds
(
self
):
...
...
tests/models/cpmant/test_modeling_cpmant.py
View file @
609e6622
...
...
@@ -151,12 +151,7 @@ class CpmAntModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
CpmAntConfig
)
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_common_properties
()
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
self
.
config_tester
.
run_common_tests
()
def
test_inputs_embeds
(
self
):
unittest
.
skip
(
"CPMAnt doesn't support input_embeds."
)(
self
.
test_inputs_embeds
)
...
...
tests/models/cvt/test_modeling_cvt.py
View file @
609e6622
...
...
@@ -162,15 +162,16 @@ class CvtModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
def
setUp
(
self
):
self
.
model_tester
=
CvtModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
CvtConfig
,
has_text_modality
=
False
,
hidden_size
=
37
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
CvtConfig
,
has_text_modality
=
False
,
hidden_size
=
37
,
common_properties
=
[
"hidden_size"
,
"num_channels"
],
)
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
self
.
config_tester
.
run_common_tests
()
@
unittest
.
skip
(
reason
=
"Cvt does not output attentions"
)
def
test_attention_outputs
(
self
):
...
...
tests/models/deformable_detr/test_modeling_deformable_detr.py
View file @
609e6622
...
...
@@ -230,15 +230,15 @@ class DeformableDetrModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineT
def
setUp
(
self
):
self
.
model_tester
=
DeformableDetrModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
DeformableDetrConfig
,
has_text_modality
=
False
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
DeformableDetrConfig
,
has_text_modality
=
False
,
common_properties
=
[
"num_channels"
,
"d_model"
,
"encoder_attention_heads"
,
"decoder_attention_heads"
],
)
def
test_config
(
self
):
# we don't test common_properties and arguments_init as these don't apply for Deformable DETR
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
run_common_tests
()
def
test_deformable_detr_model
(
self
):
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
()
...
...
tests/models/depth_anything/test_modeling_depth_anything.py
View file @
609e6622
...
...
@@ -149,17 +149,15 @@ class DepthAnythingModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.Tes
def
setUp
(
self
):
self
.
model_tester
=
DepthAnythingModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
DepthAnythingConfig
,
has_text_modality
=
False
,
hidden_size
=
37
self
,
config_class
=
DepthAnythingConfig
,
has_text_modality
=
False
,
hidden_size
=
37
,
common_properties
=
[
"patch_size"
],
)
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained_subfolder
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
self
.
config_tester
.
run_common_tests
()
@
unittest
.
skip
(
reason
=
"Depth Anything with AutoBackbone does not have a base model and hence no input_embeddings"
)
def
test_inputs_embeds
(
self
):
...
...
tests/models/dinat/test_modeling_dinat.py
View file @
609e6622
...
...
@@ -219,15 +219,12 @@ class DinatModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
def
setUp
(
self
):
self
.
model_tester
=
DinatModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
DinatConfig
,
embed_dim
=
37
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
DinatConfig
,
embed_dim
=
37
,
common_properties
=
[
"patch_size"
,
"num_channels"
]
)
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
self
.
config_tester
.
run_common_tests
()
def
test_model
(
self
):
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
()
...
...
tests/models/donut/test_modeling_donut_swin.py
View file @
609e6622
...
...
@@ -153,15 +153,16 @@ class DonutSwinModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCas
def
setUp
(
self
):
self
.
model_tester
=
DonutSwinModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
DonutSwinConfig
,
embed_dim
=
37
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
DonutSwinConfig
,
has_text_modality
=
False
,
embed_dim
=
37
,
common_properties
=
[
"image_size"
,
"patch_size"
,
"num_channels"
],
)
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
self
.
config_tester
.
run_common_tests
()
def
test_model
(
self
):
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
()
...
...
tests/models/efficientnet/test_modeling_efficientnet.py
View file @
609e6622
...
...
@@ -142,16 +142,15 @@ class EfficientNetModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.Test
def
setUp
(
self
):
self
.
model_tester
=
EfficientNetModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
EfficientNetConfig
,
has_text_modality
=
False
,
hidden_size
=
37
self
,
config_class
=
EfficientNetConfig
,
has_text_modality
=
False
,
hidden_size
=
37
,
common_properties
=
[
"num_channels"
,
"image_size"
,
"hidden_dim"
],
)
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
self
.
config_tester
.
run_common_tests
()
@
unittest
.
skip
(
reason
=
"EfficientNet does not use inputs_embeds"
)
def
test_inputs_embeds
(
self
):
...
...
tests/models/focalnet/test_modeling_focalnet.py
View file @
609e6622
...
...
@@ -250,15 +250,16 @@ class FocalNetModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase
def
setUp
(
self
):
self
.
model_tester
=
FocalNetModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
FocalNetConfig
,
embed_dim
=
37
,
has_text_modality
=
False
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
FocalNetConfig
,
embed_dim
=
37
,
has_text_modality
=
False
,
common_properties
=
[
"image_size"
,
"patch_size"
,
"num_channels"
,
"hidden_sizes"
],
)
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
self
.
config_tester
.
run_common_tests
()
def
test_model
(
self
):
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
()
...
...
tests/models/grounding_dino/test_modeling_grounding_dino.py
View file @
609e6622
...
...
@@ -246,15 +246,15 @@ class GroundingDinoModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.Tes
def
setUp
(
self
):
self
.
model_tester
=
GroundingDinoModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
GroundingDinoConfig
,
has_text_modality
=
False
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
GroundingDinoConfig
,
has_text_modality
=
False
,
common_properties
=
[
"d_model"
,
"encoder_attention_heads"
,
"decoder_attention_heads"
],
)
def
test_config
(
self
):
# we don't test common_properties and arguments_init as these don't apply for Grounding DINO
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
run_common_tests
()
def
test_model
(
self
):
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
()
...
...
tests/models/levit/test_modeling_levit.py
View file @
609e6622
...
...
@@ -190,15 +190,12 @@ class LevitModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
def
setUp
(
self
):
self
.
model_tester
=
LevitModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
LevitConfig
,
has_text_modality
=
False
,
hidden_size
=
37
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
LevitConfig
,
has_text_modality
=
False
,
common_properties
=
[
"image_size"
,
"num_channels"
]
)
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
self
.
config_tester
.
run_common_tests
()
@
unittest
.
skip
(
reason
=
"Levit does not use inputs_embeds"
)
def
test_inputs_embeds
(
self
):
...
...
tests/models/maskformer/test_modeling_maskformer_swin.py
View file @
609e6622
...
...
@@ -184,7 +184,13 @@ class MaskFormerSwinModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.Te
def
setUp
(
self
):
self
.
model_tester
=
MaskFormerSwinModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
MaskFormerSwinConfig
,
embed_dim
=
37
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
MaskFormerSwinConfig
,
has_text_modality
=
False
,
embed_dim
=
37
,
common_properties
=
[
"image_size"
,
"patch_size"
,
"num_channels"
],
)
@
require_torch_multi_gpu
@
unittest
.
skip
(
...
...
@@ -197,12 +203,7 @@ class MaskFormerSwinModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.Te
pass
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
self
.
config_tester
.
run_common_tests
()
def
test_model
(
self
):
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
()
...
...
tests/models/perceiver/test_modeling_perceiver.py
View file @
609e6622
...
...
@@ -315,7 +315,12 @@ class PerceiverModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCas
def
setUp
(
self
):
self
.
model_tester
=
PerceiverModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
PerceiverConfig
,
hidden_size
=
37
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
PerceiverConfig
,
hidden_size
=
37
,
common_properties
=
[
"d_model"
,
"num_self_attention_heads"
,
"num_cross_attention_heads"
],
)
def
_prepare_for_class
(
self
,
inputs_dict
,
model_class
,
return_labels
=
False
):
inputs_dict
=
copy
.
deepcopy
(
inputs_dict
)
...
...
@@ -344,12 +349,7 @@ class PerceiverModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCas
return
inputs_dict
def
test_config
(
self
):
# we don't test common_properties and arguments_init as these don't apply for Perceiver
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
run_common_tests
()
def
test_for_masked_lm
(
self
):
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
(
model_class
=
PerceiverForMaskedLM
)
...
...
tests/models/regnet/test_modeling_regnet.py
View file @
609e6622
...
...
@@ -136,15 +136,15 @@ class RegNetModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
def
setUp
(
self
):
self
.
model_tester
=
RegNetModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
RegNetConfig
,
has_text_modality
=
False
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
RegNetConfig
,
has_text_modality
=
False
,
common_properties
=
[
"num_channels"
,
"hidden_sizes"
],
)
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
self
.
config_tester
.
run_common_tests
()
@
unittest
.
skip
(
reason
=
"RegNet does not use inputs_embeds"
)
def
test_inputs_embeds
(
self
):
...
...
tests/models/resnet/test_modeling_resnet.py
View file @
609e6622
...
...
@@ -181,15 +181,15 @@ class ResNetModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
def
setUp
(
self
):
self
.
model_tester
=
ResNetModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
ResNetConfig
,
has_text_modality
=
False
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
ResNetConfig
,
has_text_modality
=
False
,
common_properties
=
[
"num_channels"
,
"hidden_sizes"
],
)
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
self
.
config_tester
.
run_common_tests
()
@
unittest
.
skip
(
reason
=
"ResNet does not use inputs_embeds"
)
def
test_inputs_embeds
(
self
):
...
...
Prev
1
2
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment