mmpretrain/tests/test_models
Yixiao Fang 0826df8963
[Feature] Add ViT of SAM (#1476)
* add vit of sam

* update

* update

* add ut

* update ut

* remove num_classes

* support dynamic input

* add ut

* add comments

* update ut
2023-04-13 17:03:28 +08:00
..
test_backbones update riformer mmpretrain 2023-04-06 21:56:25 +08:00
test_selfsup [Refactor] Use `out_type` to specify ViT-like backbone output. (#1408) 2023-03-09 11:02:58 +08:00
test_utils [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
test_classifiers.py [Refactor] Refactor ClsDatasample to a union DataSample. (#1371) 2023-02-23 10:07:53 +08:00
test_heads.py [Refactor] Refactor ClsDatasample to a union DataSample. (#1371) 2023-02-23 10:07:53 +08:00
test_losses.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00
test_models.py [Feature] Add ViT of SAM (#1476) 2023-04-13 17:03:28 +08:00
test_necks.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00
test_retrievers.py [Refactor] Move transforms in mmselfsup to mmpretrain. (#1396) 2023-03-03 15:01:11 +08:00
test_tta.py [Refactor] Refactor ClsDatasample to a union DataSample. (#1371) 2023-02-23 10:07:53 +08:00