Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
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
Show 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 = {
}
class
Dinov2Config
(
PretrainedConfig
,
BackboneConfigMixin
):
class
Dinov2Config
(
BackboneConfigMixin
,
PretrainedConfig
):
r
"""
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
...
...
tests/models/timm_backbone/test_modeling_timm_backbone.py
View file @
de8bec6d
...
...
@@ -106,8 +106,9 @@ class TimmBackboneModelTest(ModelTesterMixin, BackboneTesterMixin, PipelineTeste
has_attentions
=
False
def
setUp
(
self
):
self
.
config_class
=
PretrainedConfig
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
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
...
...
tests/test_backbone_common.py
View file @
de8bec6d
...
...
@@ -15,6 +15,7 @@
import
copy
import
inspect
import
tempfile
from
transformers.testing_utils
import
require_torch
,
torch_device
from
transformers.utils.backbone_utils
import
BackboneType
...
...
@@ -72,6 +73,16 @@ class BackboneTesterMixin:
expected_arg_names
=
[
"pixel_values"
]
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
):
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