Commit Graph

69 Commits (7ed40fb9a709a1560facc004147a8ccb037beba2)

Author SHA1 Message Date
Yang Nie 4cfd2159e5 rename cvt_{depth}_{size}x{size} to CvT_{depth}_{size} 2023-04-04 18:44:44 +08:00
Yang Nie d7a1127559 add CvT 2023-04-04 18:44:44 +08:00
gaotingquan 5c39dfa6ba rename gvt.py -> twins.py & twins-svt -> twins-alt-gvt 2023-03-30 17:29:49 +08:00
tianyi1997 c3fa6eca86 Create backbone for MetaBIN 2023-02-28 15:01:21 +08:00
zh-hike 23f5af9f2a add field vit to foundationvit's name 2023-02-07 17:38:50 +08:00
zh-hike e8bf35d1b4 fix foundationvit load fail 2023-02-07 10:46:25 +08:00
zhangyubo0722 a170400d20 modify init 2023-02-01 10:43:11 +08:00
zhangyubo0722 79cbd7350a Aesthetic 2023-02-01 10:43:11 +08:00
HydrogenSulfate 3a8b56801d feat(model): add EfficientNetV2 code and fix AttrDict BUG 2023-01-31 21:15:22 +08:00
zhangyubo0722 54f7dd7484 watermark 2023-01-31 19:57:43 +08:00
zhangyubo0722 272e75365b modify swinv2 2023-01-20 14:08:06 +08:00
zhangyubo0722 65e7a41a1d add docs 2023-01-20 14:08:06 +08:00
zhangyubo0722 7e616a16b5 modify init 2023-01-20 14:08:06 +08:00
zhangyubo0722 f5e8df0e63 Some modifications 2023-01-20 14:08:06 +08:00
zhangyubo0722 da95893bf2 supplement init 2023-01-20 14:08:06 +08:00
LittleMoon e069dedd7e modified DSNet model names 2022-12-22 14:55:46 +08:00
LittleMoon 28e094e097 add docs, config, model for DSNet 2022-12-22 14:55:46 +08:00
zhangyubo0722 e238686e84 modify init 2022-12-22 14:54:25 +08:00
zhangyubo0722 0140b38a80 add models 2022-12-22 14:54:25 +08:00
zhangyubo0722 b61da414e5 add uniformer lst 2022-12-02 19:45:59 +08:00
cuicheng01 ec2029b821 Add NextViT code and docs 2022-11-23 14:59:55 +08:00
gaotingquan cfab4c178a feat: add pplcnetv2_small and pplcnetv2_big 2022-11-18 13:14:37 +08:00
dongshuilong 944763d7a5 add fixmatch 2022-10-25 12:04:22 +08:00
HydrogenSulfate 3647da6d4a
Merge pull request #2383 from HydrogenSulfate/xbm_final
Add XBM model
2022-10-19 15:43:40 +08:00
gaotingquan ad3c657e39 fix: update VAN_tiny to VAN_B0 2022-10-18 17:48:31 +08:00
HydrogenSulfate 2cfd8dd828 add xbm 2022-10-17 15:45:45 +08:00
zengshao0622 90d591e410 update ppcls 2022-09-08 11:34:54 +00:00
zengshao0622 7b50ce6585 merge CAE 2022-09-08 08:11:25 +00:00
HydrogenSulfate 1b5e00e82a add PP-ShiTuV2 code 2022-08-23 07:54:58 +00:00
root 2a909c306a fix: change to relative import 2022-08-19 11:40:38 +08:00
cuicheng01 1fccb35be7
Merge branch 'develop' into PeleeNet_PR 2022-07-14 10:47:06 +08:00
Yang Nie 4266b8790d rename `ConvNext` to `ConvNeXt` 2022-07-09 23:25:50 +08:00
Yang Nie 436f64ad84 Merge branch 'develop' into ConvNeXt 2022-07-09 21:50:31 +08:00
Walter dcef409dde
Merge branch 'develop' into adaface 2022-05-27 14:19:03 +08:00
littletomatodonkey 6dac07f68d
fix quant logic (#1941)
* fix quant logic

* add support for trt+int8 inference

* add support for swin quant

* fix swin and quant

* fix assert info

* fix assert info

* fix log
2022-05-23 19:04:27 +08:00
dongshuilong 759fe41ccc modify face_dataset and ir_net name issue 2022-05-19 02:22:17 +00:00
cuicheng01 03142ea32b
Merge pull request #1919 from cuicheng01/develop
Add PPHGNet code
2022-05-13 17:59:53 +08:00
cuicheng01 8a760fb85f Add PPHGNet code 2022-05-12 15:53:51 +00:00
Yang Nie 96687d07e2 add PeleeNet code 2022-05-12 23:19:44 +08:00
gaotingquan 925cb6ef6c
feat: add PPLCNetV2 2022-05-12 14:38:17 +00:00
dongshuilong bdc535bb39 add adaface 2022-05-06 04:50:26 +00:00
flytocc e1943f9a68 add ConvNeXt code 2022-04-28 00:34:48 +08:00
flytocc c3018ebd39 add VAN-Classification code 2022-04-19 13:13:33 +08:00
cuicheng01 44b9963b25 add mobilevit code 2022-04-01 08:51:42 +00:00
cuicheng01 51e06d43f5 add cswin code and doc 2022-03-29 09:50:44 +00:00
Tingquan Gao 158c7b4e35 fix: fix bug that overrided eval() can't be called in distributed
The eval() overrided can't be called because model is a DataParallel object in distributed.
2022-03-04 16:52:57 +08:00
gaotingquan 56c72a85a1 fix: fix lr 2022-01-07 11:52:30 +08:00
gaotingquan a53b636829 feat: support PVTV2 2022-01-07 11:52:30 +08:00
cuicheng01 897760f073
Merge pull request #1544 from cuicheng01/update_vit
update vit
2021-12-14 14:40:14 +08:00
cuicheng01 83d1b411c5 update vit 2021-12-10 09:56:33 +00:00