Commit Graph

272 Commits (465a8e5fd32633b25eba3faedee7d78a522024aa)

Author SHA1 Message Date
cuicheng01 465a8e5fd3
Create resnet.py 2021-05-25 19:18:14 +08:00
weishengyu 2b150a2ffa dbg 2021-05-25 16:51:04 +08:00
weishengyu 88f92297fa dbg 2021-05-25 16:50:37 +08:00
weishengyu 79e9419fad dbg 2021-05-25 16:47:33 +08:00
weishengyu 611a40958a dbg 2021-05-25 16:40:20 +08:00
weishengyu 1499b0e585 add nn 2021-05-25 15:50:50 +08:00
weishengyu 6520a80176 add res_dict 2021-05-25 15:46:41 +08:00
weishengyu 1406b55aba add hrnet 2021-05-25 15:20:07 +08:00
weishengyu 010521e828 add folder 2021-05-25 14:45:50 +08:00
weishengyu 8f3df96eea move files 2021-05-25 14:39:53 +08:00
weishengyu 608b163d4f move stop_after out of update_sub, and rename update_sub to update_res 2021-05-24 19:49:06 +08:00
weishengyu f6da3bd60a move files 2021-05-24 11:42:24 +08:00
guofei 86346fda54
Support AMP for SE_RexNeXt101_32x4d (#681) 2021-05-21 12:50:49 +08:00
Yiqun Liu 0d832a2539
Enable profiler, for both static and dynamic training. (#729)
* Enable profiler for static training.

* Polish the initialize of ProfilerOptions.

* Enable profiler for dynamic mode.
2021-05-21 10:31:43 +08:00
jm12138 6360a61e59
rm label in forword func (#731) 2021-05-18 21:51:15 +08:00
Tingquan Gao e5f7529587
Add SwinTransformer (#721)
* Add SwinTransformer_large

* Add SwinTransformer

* Optimize user experience

* Add the doc of SwinTransformer

* Modify the format of the corner mark
2021-05-17 14:55:38 +08:00
liuyuhui 5170153ee3 merge with upstream/develop 2021-05-08 11:16:41 +00:00
liuyuhui 70f1782339 fix xpu reader 2021-05-08 07:15:38 +00:00
littletomatodonkey e165897c37
fix drop last for training process (#713) 2021-05-06 14:21:54 +08:00
littletomatodonkey 9c0f049603
fix export model eval (#710)
* adapt to net.eval for the framework just contains training flag setting
* fix bug when export swin transformer
2021-05-06 13:55:11 +08:00
littletomatodonkey b480d92c17
add swin transformer (#685)
* add swin transformer
2021-04-27 15:14:56 +08:00
cuicheng01 6c2a33fe51
fix warnings bug (#698) 2021-04-25 18:48:26 +08:00
gaotingquan e8435f6145 Fix raise exception when overriding params 2021-04-16 10:55:31 +08:00
cuicheng01 d6e41d7b17
Merge pull request #686 from RainFrost1/add_mixnet_rexnet
add_mixnet_rexnet
2021-04-15 19:08:57 +08:00
dongshuilong 2e14f726d0 fix mixnet review problems 2021-04-15 17:15:18 +08:00
dongshuilong ea63084670 add_mixnet_rexnet 2021-04-15 14:05:46 +08:00
littletomatodonkey a7aa14525c
fix repvgg eval (#677)
* fix repvgg eval

* fix dp training

* fix single card train
2021-04-14 02:16:13 +08:00
littletomatodonkey b0fa8dcea9
fix log when running vit in static mode (#654)
* fix log when running vit in statich mode

* fix comment
2021-04-01 17:36:59 +08:00
cuicheng01 23104797a6
Merge pull request #656 from flyseaworld/develop
add multilabel feature based develop
2021-04-01 15:52:32 +08:00
huangxu96 525b5e1a0e
Support fp16 training for ResNeXt101_32x4d (#653) 2021-03-31 11:10:19 +08:00
yaohai fee32b555a fix small error 2021-03-30 17:14:25 +08:00
yaohai 5fd7085ddf add multilabel feature 2021-03-30 16:02:32 +08:00
Tingquan Gao 8832a3fa0a
Support Visual DL (#650)
* Support Visual DL

* Fix VDL

* Add doc of VDL, test=document_fix

* Add the en doc of VDL, test=document_fix

Co-authored-by: littletomatodonkey <2120160898@bit.edu.cn>
2021-03-23 12:45:01 +08:00
littletomatodonkey bba176e05a
enhance reader (#628) 2021-03-08 10:35:38 +08:00
huangxu96 1df6641853
Merge branch 'develop' into cp_fp16_training 2021-03-01 20:01:20 +08:00
littletomatodonkey e02a35ac45
add quant clas (#621)
* add quant clas

* fix load init model

* add quant

* add quant doc

* fix export model
2021-03-01 19:36:55 +08:00
huangxu96 4e43ec6995 new usage of amp training. (#564)
* new usage of amp training.

* change the usage of amp and pure fp16 training.

* modified code as reviews
2021-02-26 09:25:54 +00:00
littletomatodonkey 0c35c4c794
fix static run for vit and deit (#605) 2021-02-06 23:10:38 +08:00
littletomatodonkey a6667a15ab
Update __init__.py 2021-02-02 23:26:13 +08:00
jm12138 8cd779be4d
Add RepVGG models (#592)
* Add RepVGG models

* Update model code

* Rename code

* format code

* add README of RepVGG
2021-02-01 18:07:25 +08:00
littletomatodonkey 378ed96095
fix mv2 prefix name (#591) 2021-01-31 18:18:40 +08:00
littletomatodonkey a40d348ebb
Update mobilenet_v3.py 2021-01-26 11:57:37 +08:00
littletomatodonkey 9ebbd78bc9
fix save distillation model (#578)
* fix save distillation model

* add note
2021-01-25 20:58:35 +08:00
jm12138 c07c5de1ed
Fix a bug of DropPath (#575) 2021-01-25 11:00:43 +08:00
jm12138 1b89b8a58a
Add the DeiT models (#573)
* Add the DeiT models

* fix a training bug

* update the model init config

* update the License
2021-01-22 15:30:55 +08:00
jm12138 747a659891
Add ViT model (#570)
* Add the ViT model
2021-01-21 13:05:05 +08:00
littletomatodonkey d0ecff1b5a
fix save distillation model (#567) 2021-01-19 18:49:30 +08:00
littletomatodonkey 0e6fe6f1cf
refine distillation model (#565) 2021-01-19 13:54:27 +08:00
littletomatodonkey 223a0d2e7e
fix vgg stop grad (#558)
* fix vgg stop grad

* beautify code
2021-01-14 19:48:41 +08:00
littletomatodonkey 8e26493c32
fix lr mult val (#551) 2021-01-07 19:24:29 +08:00