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
de8bec6d
Unverified
Commit
de8bec6d
authored
Sep 18, 2023
by
NielsRogge
Committed by
GitHub
Sep 18, 2023
Browse files
[AutoBackbone] Add test (#26094)
* Add test * Add config_class
parent
97f439ae
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
14 additions
and
2 deletions
+14
-2
src/transformers/models/dinov2/configuration_dinov2.py
src/transformers/models/dinov2/configuration_dinov2.py
+1
-1
tests/models/timm_backbone/test_modeling_timm_backbone.py
tests/models/timm_backbone/test_modeling_timm_backbone.py
+2
-1
tests/test_backbone_common.py
tests/test_backbone_common.py
+11
-0
No files found.
src/transformers/models/dinov2/configuration_dinov2.py
View file @
de8bec6d
...
@@ -32,7 +32,7 @@ DINOV2_PRETRAINED_CONFIG_ARCHIVE_MAP = {
...
@@ -32,7 +32,7 @@ DINOV2_PRETRAINED_CONFIG_ARCHIVE_MAP = {
}
}
class
Dinov2Config
(
PretrainedConfig
,
BackboneConfigMixin
):
class
Dinov2Config
(
BackboneConfigMixin
,
PretrainedConfig
):
r
"""
r
"""
This is the configuration class to store the configuration of a [`Dinov2Model`]. It is used to instantiate an
This is the configuration class to store the configuration of a [`Dinov2Model`]. It is used to instantiate an
Dinov2 model according to the specified arguments, defining the model architecture. Instantiating a configuration
Dinov2 model according to the specified arguments, defining the model architecture. Instantiating a configuration
...
...
tests/models/timm_backbone/test_modeling_timm_backbone.py
View file @
de8bec6d
...
@@ -106,8 +106,9 @@ class TimmBackboneModelTest(ModelTesterMixin, BackboneTesterMixin, PipelineTeste
...
@@ -106,8 +106,9 @@ class TimmBackboneModelTest(ModelTesterMixin, BackboneTesterMixin, PipelineTeste
has_attentions
=
False
has_attentions
=
False
def
setUp
(
self
):
def
setUp
(
self
):
self
.
config_class
=
PretrainedConfig
self
.
model_tester
=
TimmBackboneModelTester
(
self
)
self
.
model_tester
=
TimmBackboneModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
PretrainedConfig
,
has_text_modality
=
False
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
self
.
config_class
,
has_text_modality
=
False
)
def
test_config
(
self
):
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_string
()
...
...
tests/test_backbone_common.py
View file @
de8bec6d
...
@@ -15,6 +15,7 @@
...
@@ -15,6 +15,7 @@
import
copy
import
copy
import
inspect
import
inspect
import
tempfile
from
transformers.testing_utils
import
require_torch
,
torch_device
from
transformers.testing_utils
import
require_torch
,
torch_device
from
transformers.utils.backbone_utils
import
BackboneType
from
transformers.utils.backbone_utils
import
BackboneType
...
@@ -72,6 +73,16 @@ class BackboneTesterMixin:
...
@@ -72,6 +73,16 @@ class BackboneTesterMixin:
expected_arg_names
=
[
"pixel_values"
]
expected_arg_names
=
[
"pixel_values"
]
self
.
assertListEqual
(
arg_names
[:
1
],
expected_arg_names
)
self
.
assertListEqual
(
arg_names
[:
1
],
expected_arg_names
)
def
test_config_save_pretrained
(
self
):
config_class
=
self
.
config_class
config_first
=
config_class
(
out_indices
=
[
0
,
1
,
2
,
3
])
with
tempfile
.
TemporaryDirectory
()
as
tmpdirname
:
config_first
.
save_pretrained
(
tmpdirname
)
config_second
=
self
.
config_class
.
from_pretrained
(
tmpdirname
)
self
.
assertEqual
(
config_second
.
to_dict
(),
config_first
.
to_dict
())
def
test_channels
(
self
):
def
test_channels
(
self
):
config
,
_
=
self
.
model_tester
.
prepare_config_and_inputs_for_common
()
config
,
_
=
self
.
model_tester
.
prepare_config_and_inputs_for_common
()
...
...
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