Commit Graph

34 Commits (451b0a3cce00b6eec4effa7c0fbcebefb48e152c)

Author SHA1 Message Date
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
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 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
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 d868a63764 Add ESNet code and pretrained models 2021-11-11 06:00:32 +00:00
stephon 6f13c5b798 rename PPLCNet_x2_5_Tanh 2021-11-04 08:37:21 +00:00
stephon df03b4a022 add Binary general reocg configure 2021-11-03 10:09:34 +00:00
littletomatodonkey f8b420946c
fix cspnet (#1282)
* fix cspnet

* minor fix
2021-10-11 17:23:31 +08:00
cuicheng01 d470f34c64 Update code and docs 2021-09-13 03:18:37 +00:00
cuicheng01 4150df912e Add LCNet codes and config 2021-09-07 16:14:40 +00:00
Bin Lu f878fb27e5
Update __init__.py 2021-08-23 19:19:19 +08:00
Tingquan Gao 24727034d8
Update whl (#871)
* Update whl

* Optimize usage of CLI args
2021-06-22 11:12:26 +08:00
cuicheng01 d5a4ad19e2 Remove 2021-06-17 02:52:15 +00:00
cuicheng01 d8def84624 Update model_zoo 2021-06-11 13:58:50 +00:00
cuicheng01 f136ee1843
Merge branch 'develop_reg' into reg/add_twins_levit 2021-06-11 17:01:49 +08:00
jm12138 a950ec42b0 add the codes of TNT, HarDNet, RedNet and DLA models 2021-06-10 23:15:40 +08:00
gaotingquan 9e2a8e862a Add LeViT and Twins 2021-06-10 14:29:12 +00:00
dongshuilong 76fcde59af add variant model example 2021-06-04 10:19:01 +08:00
cuicheng01 d13cb22378 Add legendary_model's congigs and fix some trainer's bug 2021-06-01 17:24:21 +00:00
weishengyu 8f3df96eea move files 2021-05-25 14:39:53 +08:00
weishengyu f6da3bd60a move files 2021-05-24 11:42:24 +08:00