Commit Graph

211 Commits (0146af28b03c3e0610894b45b6b3d12333a62ca5)

Author SHA1 Message Date
flytocc 7b020eb2aa add paper citation 2022-04-19 19:45:19 +08:00
flytocc c3018ebd39 add VAN-Classification code 2022-04-19 13:13:33 +08:00
cuicheng01 96fbbad038 update mobilevit.py 2022-04-01 09:06:46 +00:00
cuicheng01 44b9963b25 add mobilevit code 2022-04-01 08:51:42 +00:00
cuicheng01 4839915056 update cswin_transformer.py 2022-03-29 12:01:04 +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
Wei Shengyu 1020f41899
use flatten instead of reshape for hrnet (#1709) 2022-02-28 19:12:53 +08:00
WangChen0902 7595ba6d70
add AFD (#1683)
* add AFD
2022-02-28 19:11:50 +08:00
gaotingquan 7abf42b7a1 fix: rm some repvgg models not supported
remove RepVGG_B3, RepVGG_B2g2 and RepVGG_B3g2 that are not supported by PaddleClas.
2022-02-25 23:50:29 +08:00
gaotingquan 4619d63273 fix: fix the download link, etc.
fix the pretrained and inference download link;
add use_ssld arg to TNT model.
2022-02-25 23:50:29 +08:00
gaotingquan 6963c25804 fix: fix bug
bug is that error rasied by calling update_res() when sublayer does not have _save_sub_res_hook() because nn.Sequential and nn.LayerList is not TheseusLayer.
2022-01-14 14:58:28 +08:00
gaotingquan a6a76f3d36 fix: args error 2022-01-14 14:58:28 +08:00
gaotingquan 43fce425ac fix: change the returned result from dict to list 2022-01-14 14:58:28 +08:00
gaotingquan 3b9f629220 feat: support specify return_stages to update_res() 2022-01-14 14:58:28 +08:00
gaotingquan 56c72a85a1 fix: fix lr 2022-01-07 11:52:30 +08:00
gaotingquan 025feb6de9 fix: fix -1 in dims 2022-01-07 11:52:30 +08:00
gaotingquan ee819393ac fix: fix -1 in dims and ignore swapdim in static 2022-01-07 11:52:30 +08:00
gaotingquan 00fb3f7519 feat: support PVTV2 2022-01-07 11:52:30 +08:00
gaotingquan a53b636829 feat: support PVTV2 2022-01-07 11:52:30 +08:00
littletomatodonkey aea712cc87
add dist of rec model (#1574)
* add distillation loss func and rec distillation
2022-01-05 19:25:36 +08:00
gaotingquan d6aeba6f56
fix: upgrade replace_sub() to upgrade_sublayer()
supplement to PR #1581
2021-12-30 10:59:20 +00:00
gaotingquan 71cb728b07 fix: rename to upgrade_sublayer() 2021-12-28 15:24:50 +08:00
gaotingquan 1e696ac22b fix: remove unnecessary register_hook() call & pre-commit 2021-12-28 15:24:50 +08:00
gaotingquan a86c4b290e add license 2021-12-28 15:24:50 +08:00
gaotingquan 721ac0bf61 refactor: simplify code
1. remove WrapLayer and wrap_theseus;
2. support call update_res() one more;
3. optim parse_pattern_str() to return list of layer parsed.
2021-12-28 15:24:50 +08:00
gaotingquan 56911b573b refactor: rename replace_sub() func to ? 2021-12-28 15:24:50 +08:00
gaotingquan 0f126b75da refactor: strengthen parse_pattern_str() func 2021-12-28 15:24:50 +08:00
gaotingquan 8d0b0d4b0a refactor: extract _parse_pattern_str() func 2021-12-28 15:24:50 +08:00
gaotingquan 18dec0744a fix: fix problems commented in reviewing 2021-12-28 15:24:50 +08:00
gaotingquan 41296972a4 fix: fix comments 2021-12-28 15:24:50 +08:00
gaotingquan f8ee6c0f86 fix: fix result returned by stop_after 2021-12-28 15:24:50 +08:00
gaotingquan cf205e1379 fix: fix result returned by _find_layers_handle 2021-12-28 15:24:50 +08:00
gaotingquan b0ae3a1210 fix: unify the pattern of layer's name 2021-12-28 15:24:50 +08:00
cuicheng01 897760f073
Merge pull request #1544 from cuicheng01/update_vit
update vit
2021-12-14 14:40:14 +08:00
cuicheng01 b3922c96f5
Merge pull request #1436 from TingquanGao/dev/update_whl
update whl
2021-12-14 12:15:46 +08:00
cuicheng01 83d1b411c5 update vit 2021-12-10 09:56:33 +00:00
weishengyu 6c5d1ebc28 add pruner and quanter for theseus 2021-12-09 14:51:40 +08:00
cuicheng01 5dad991c2a update se_resnext_vd.py 2021-11-30 15:06:05 +00:00
gaotingquan d53a06c625 fix: fix download link 2021-11-17 15:02:20 +08:00
gaotingquan ae6cb7d529 feat: support MixNet 2021-11-15 06:53:34 +00:00
cuicheng01 6e4bf593fb
Merge pull request #1390 from Intsigstephon/feature_binary_model
add Binary general recog configure
2021-11-12 15:25:00 +08:00
cuicheng01 fb631b38fb update esnet.py 2021-11-11 12:51:58 +00:00
cuicheng01 d868a63764 Add ESNet code and pretrained models 2021-11-11 06:00:32 +00:00
littletomatodonkey 67d982c30e
fix csp refer (#1404) 2021-11-05 13:54:10 +08:00
cuicheng01 7875e83d9e update vision_transformer.py 2021-11-05 02:53:10 +00:00
cuicheng01 08952deb88 update codes 2021-11-04 10:47:13 +00:00
Bin Lu 9f2f62f3ae
Update pp_lcnet_variant.py 2021-11-04 16:58:17 +08:00
Bin Lu 17fc4595b3
Create pp_lcnet_variant.py 2021-11-04 16:44:56 +08:00
Bin Lu cad5ce842a
Update __init__.py 2021-11-04 16:44:16 +08:00