Commit Graph

5 Commits (0826df8963915a0389178e30808fc1bceecab218)

Author SHA1 Message Date
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
techmonsterwang 32c258ff19 update riformer mmpretrain 2023-04-06 21:56:25 +08:00
techmonsterwang 0b70c108b0 update riformer mmpretrain 2023-04-06 21:56:25 +08:00
Yixiao Fang e453a45d31
[Refactor] Add self-supervised backbones and target generators. (#1379)
* add heads

* add losses

* fix

* remove mim head

* add modified backbones and target generators

* add unittest

* refactor caevit

* add window_size check

* fix lint

* apply new DataSample

* fix ut error

* update ut

* fix ut

* fix lint

* Update base modules.

---------

Co-authored-by: mzr1996 <mzr1996@163.com>
2023-02-28 15:59:17 +08:00
mzr1996 0979e78573 Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00