SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Add ViTDet (#25524)

Browse Source 
* First draft

* Fix READMEs

* Update return_dict

* Add more tests

* Fix docstrings

* Address comments

* Address more comments

* Address more comments

* Address more comments, fix test

* Fix test
This commit is contained in:
NielsRogge
2023-08-29 11:03:52 +02:00
committed by GitHub
parent 99c3d44906
commit 4c21da5e34
22 changed files with 1477 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_backbone_common.py
View File

@@ -31,7 +31,8 @@ class BackboneTesterMixin:
# test default config
config = config_class()
self.assertIsNotNone(config)
expected_stage_names = ["stem"] + [f"stage{idx}" for idx in range(1, len(config.depths) + 1)]
num_stages = len(config.depths) if hasattr(config, "depths") else config.num_hidden_layers
expected_stage_names = ["stem"] + [f"stage{idx}" for idx in range(1, num_stages + 1)]
self.assertEqual(config.stage_names, expected_stage_names)
self.assertTrue(set(config.out_features).issubset(set(config.stage_names)))