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