Commit Graph

577 Commits (b63cad07c0d1b95505caa19fdce03982a03873f6)

Author SHA1 Message Date
littletomatodonkey 0a0e34298b
add support for infer and export (#760)
* add support for infer

* add export model support

* fix yaml

* fix post process name

* fix topk name
2021-06-01 14:37:47 +08:00
Bin Lu 2eb2c26907
Update ResNet50_retrieval.yml 2021-06-01 14:24:48 +08:00
Bin Lu 3c184362a7
Create ResNet50_retrieval.yml 2021-06-01 14:23:40 +08:00
Bin Lu fa15e53b00
Update ResNet50.yaml 2021-06-01 14:23:07 +08:00
Bin Lu ea53ad95c3
Update ResNet50.yaml 2021-06-01 11:40:17 +08:00
Bin Lu 6391c90b81
Delete ppcls/arch/neck directory 2021-06-01 11:35:44 +08:00
Bin Lu 85ca658ea2
Update __init__.py 2021-06-01 11:30:26 +08:00
Bin Lu 9c395ac802
Update __init__.py 2021-06-01 09:52:56 +08:00
Felix 46955a262d
Merge branch 'develop_reg' into develop_reg 2021-05-31 21:57:23 +08:00
Wei Shengyu 82ed94704d
Merge pull request #755 from Intsigstephon/develop_reg
add  head/neck and losses
2021-05-31 20:50:02 +08:00
Bin Lu f61a64aab6
Update arcmargin.py 2021-05-31 20:46:45 +08:00
Bin Lu 77a2c4571c
Update cosmargin.py 2021-05-31 20:46:16 +08:00
Bin Lu 6f2959af3a
Update arcmargin.py 2021-05-31 20:45:30 +08:00
Felix c9690673ca
Update multilabel_dataset.py 2021-05-31 16:49:54 +08:00
weishengyu 546d63d58a dbg 2021-05-31 16:49:25 +08:00
Felix 9c11ddca37
Update __init__.py 2021-05-31 16:47:57 +08:00
Felix b0708e6e9a
Update __init__.py 2021-05-31 16:40:45 +08:00
weishengyu c749bdf93f replace reshape with Flatten layer in ResNet 2021-05-31 16:40:33 +08:00
Felix 771449813c
Update DistributedRandomIdentitySampler.py 2021-05-31 16:01:39 +08:00
dongshuilong 2fa808517d legendary models v0.1 2021-05-31 16:01:02 +08:00
Felix bf8f2bef51
Update __init__.py 2021-05-31 16:00:32 +08:00
Felix 89a4c43187
Update random_erasing.py 2021-05-31 15:59:53 +08:00
Felix f2c0cee05c
Update randaugment.py 2021-05-31 15:59:08 +08:00
Felix ea08c644bd
Update operators.py 2021-05-31 15:58:46 +08:00
Felix 289a6fc11a
Update hide_and_seek.py 2021-05-31 15:58:16 +08:00
Felix f958e2c05e
Update grid.py 2021-05-31 15:57:54 +08:00
Felix 2f84cc3809
Update functional.py 2021-05-31 15:56:42 +08:00
Felix 017d95caf2
Update fmix.py 2021-05-31 15:55:51 +08:00
Felix f15a478fc3
Update cutout.py 2021-05-31 15:55:28 +08:00
Felix 50f00e8486
Update autoaugment.py 2021-05-31 15:54:20 +08:00
Felix 6db0ae8af4
Update batch_operators.py 2021-05-31 15:53:54 +08:00
Felix d2630c3a31
Update multilabel_dataset.py 2021-05-31 15:52:33 +08:00
Felix 56b4d8b889
Update multilabel_dataset.py 2021-05-31 15:49:57 +08:00
Felix 15fd4414d1
Update multilabel_dataset.py 2021-05-31 15:49:19 +08:00
Felix fa8dbc7f18
Update imagenet_dataset.py 2021-05-31 15:46:54 +08:00
Felix f83f6f2db8
Create common_dataset.py 2021-05-31 15:43:57 +08:00
Bin Lu 182bf94343
Update celoss.py 2021-05-31 14:28:52 +08:00
Bin Lu 8edec09dd4
Update centerloss.py 2021-05-31 14:28:40 +08:00
Bin Lu b79f3f2e65
Update comfunc.py 2021-05-31 14:28:22 +08:00
Bin Lu e9a1d83035
Update emlloss.py 2021-05-31 14:27:59 +08:00
Bin Lu adfd5c4c10
Update msmloss.py 2021-05-31 14:27:37 +08:00
Bin Lu b0d3dd9c1e
Update npairsloss.py 2021-05-31 14:27:13 +08:00
Bin Lu 36a69acb3f
Update trihardloss.py 2021-05-31 14:26:31 +08:00
Bin Lu 0efe0075a2
Update triplet.py 2021-05-31 14:25:59 +08:00
Bin Lu f62604de72
Add files via upload
add losses
2021-05-31 14:20:48 +08:00
Bin Lu 90418d79d2
Update __init__.py 2021-05-31 14:17:59 +08:00
Bin Lu bb4c2c6df0
Create __init__.py 2021-05-31 14:17:31 +08:00
Bin Lu d0edde7a94
Create circlemargin.py 2021-05-31 14:16:33 +08:00
Bin Lu 2177c3435c
Create cosmargin.py 2021-05-31 14:16:03 +08:00
Bin Lu 777f6600e0
Create fc.py 2021-05-31 14:15:37 +08:00
Bin Lu c0ea0f39bc
Create arcmargin.py 2021-05-31 14:15:14 +08:00
Bin Lu 5e064d0b24
Create __init__.py 2021-05-31 14:14:48 +08:00
Bin Lu dbb3e4bfc0
Create fc.py 2021-05-31 14:14:03 +08:00
Bin Lu 5148eabcca
Create __init__.py 2021-05-31 14:13:44 +08:00
Felix dc49f54ef8
Update random_erasing.py 2021-05-31 13:14:07 +08:00
Felix d250a69ecd
Update randaugment.py 2021-05-31 13:13:39 +08:00
Felix 7dcdd61c76
Update operators.py 2021-05-31 13:13:20 +08:00
Felix 514e3afe76
Update hide_and_seek.py 2021-05-31 13:13:03 +08:00
Felix 20bfca7522
Update grid.py 2021-05-31 13:12:46 +08:00
Felix bad16c2570
Update fmix.py 2021-05-31 13:12:27 +08:00
Felix a92060196c
Update cutout.py 2021-05-31 13:12:07 +08:00
Felix d98311ea40
Update batch_operators.py 2021-05-31 13:11:50 +08:00
Felix a6144282a1
Update autoaugment.py 2021-05-31 13:11:31 +08:00
Felix edf49437c6
Update __init__.py 2021-05-31 13:11:10 +08:00
Felix 71d139fcb7
Update random_erasing.py 2021-05-31 13:10:39 +08:00
Felix 7dd79219ea
Update randaugment.py 2021-05-31 13:10:18 +08:00
Felix ebf1b92c23
Update operators.py 2021-05-31 13:09:06 +08:00
Felix fb484d14cc
Update hide_and_seek.py 2021-05-31 13:08:50 +08:00
Felix f4d3d9ac05
Update grid.py 2021-05-31 13:08:34 +08:00
Felix c67d604985
Update fmix.py 2021-05-31 13:08:13 +08:00
Felix 41f784c1ee
Update cutout.py 2021-05-31 13:07:55 +08:00
Felix 6241ee25e1
Update batch_operators.py 2021-05-31 13:07:37 +08:00
Felix f66ae362d1
Update autoaugment.py 2021-05-31 13:06:49 +08:00
Felix dbf8df4f7c
Update __init__.py 2021-05-31 13:06:29 +08:00
Felix d32846e8c4
Update __init__.py 2021-05-31 13:05:24 +08:00
Felix 806ec9a117
Add files via upload 2021-05-31 13:01:38 +08:00
Felix d3e233defd
Create __init__.py 2021-05-31 13:00:50 +08:00
Felix 8a79cec570
Update imagenet_dataset.py 2021-05-31 12:58:54 +08:00
Felix f212345e33
Add files via upload 2021-05-31 12:58:27 +08:00
Felix 1d9c5710cc
Add files via upload 2021-05-31 12:57:18 +08:00
Felix 3bd7a0979c
Create __init__.py 2021-05-31 12:56:34 +08:00
Felix ed7e7a3717
Add files via upload 2021-05-31 12:55:28 +08:00
Felix 93ef6ed2f6
Create __init__.py 2021-05-31 12:55:10 +08:00
Felix d7ba539452
Create __init__.py 2021-05-31 12:53:48 +08:00
Walter 4d246c20e4
Merge pull request #749 from Intsigstephon/develop_reg
add mobilenet_v1.py to legendary models
2021-05-31 11:22:42 +08:00
Bin Lu c2a7f190af
Delete ppcls/arch/head directory 2021-05-31 11:17:07 +08:00
Walter e469a975ea
Merge pull request #746 from FredHuang16/patch-1
add inception_v3.py
2021-05-31 11:09:27 +08:00
Walter 8ffd18b258
Merge pull request #748 from weisy11/develop_reg
modify hrnet
2021-05-31 11:08:51 +08:00
littletomatodonkey 3881343484
add support for train and eval (#752)
* add support for train and eval

* rm unsed code

* add support for metric save and load ckp
2021-05-29 13:52:32 +08:00
Bin Lu 119f4fcb34
Update fc.py 2021-05-28 19:28:53 +08:00
Bin Lu 3ebe1e506c
Create fc.py 2021-05-28 19:23:19 +08:00
Bin Lu f7f07e6ece
Create circlemargin.py 2021-05-28 19:22:58 +08:00
Bin Lu d3351aab96
Create cosmargin.py 2021-05-28 19:22:31 +08:00
Bin Lu e3d857d4c9
Create arcmargin.py 2021-05-28 19:21:24 +08:00
Bin Lu 693aa83fe2
Create __init__.py 2021-05-28 19:21:01 +08:00
Felix 79b21af761
Update inception_v3.py 2021-05-28 18:24:18 +08:00
weishengyu 8ede57a409 add MODEL_URLS 2021-05-28 16:38:45 +08:00
Felix 4ad209e5fe
Update inception_v3.py 2021-05-28 16:20:06 +08:00
weishengyu c05df5fa89 modify code style 2021-05-28 16:06:02 +08:00
Bin Lu 53bdc09baf
Update mobilenet_v1.py 2021-05-28 14:52:32 +08:00
Bin Lu 244e632d1a
Update mobilenet_v1.py 2021-05-28 14:51:07 +08:00
Bin Lu 2b84db06c1
Update mobilenet_v1.py 2021-05-28 14:50:21 +08:00
Bin Lu 084a27a15b
Update mobilenet_v1.py 2021-05-28 14:27:37 +08:00
Bin Lu 0e7ed899eb
Update mobilenet_v1.py 2021-05-28 14:21:01 +08:00
Bin Lu d3133089bd
Update mobilenet_v1.py 2021-05-28 14:04:03 +08:00
Bin Lu aa8cf9ba04
Update mobilenet_v1.py 2021-05-28 13:33:24 +08:00
Bin Lu f5ab94b4ed
Update mobilenet_v1.py 2021-05-28 13:30:12 +08:00
Bin Lu 5d155c8fab
Update mobilenet_v1.py
add pretrained and Model_urls
2021-05-28 13:28:11 +08:00
Walter decdb51bb0
Merge pull request #742 from cuicheng01/develop_reg
Add resnet.py in legendary_models
2021-05-28 11:36:24 +08:00
cuicheng01 0f3d321b81
Update resnet.py 2021-05-28 11:34:39 +08:00
cuicheng01 e6792d34da
Update resnet.py 2021-05-28 11:32:51 +08:00
cuicheng01 90321ce38d
Update resnet.py 2021-05-28 11:29:37 +08:00
cuicheng01 45b02b86af
Update resnet.py 2021-05-28 10:57:26 +08:00
cuicheng01 31e59dfa1b
add load_dygraph_pretrain_from_url function 2021-05-28 10:55:48 +08:00
cuicheng01 5671f9d9bb
Update resnet.py 2021-05-28 10:50:17 +08:00
littletomatodonkey 487c797230
fix optimizer builder (#751) 2021-05-28 10:39:51 +08:00
Bin Lu 99adcce914
Update mobilenet_v1.py 2021-05-28 10:38:00 +08:00
Bin Lu 86eb2bc693
Update mobilenet_v1.py 2021-05-28 10:36:48 +08:00
Bin Lu 5f3456767d
Create mobilenet_v1.py 2021-05-27 19:13:03 +08:00
littletomatodonkey dd79f81fd7
[WIP]add arch init (#744)
* polish trainer
2021-05-27 18:41:44 +08:00
weishengyu dfe7f5b0ed dbg 2021-05-27 17:03:58 +08:00
weishengyu 014f1d1d27 add comment and change args of HRNet 2021-05-27 17:02:28 +08:00
Bin Lu f0e988e820
Update mobilenet_v1.py 2021-05-27 16:22:52 +08:00
Felix b8ac9ee31b
Update inception_v3.py 2021-05-27 14:54:30 +08:00
Felix 670c2bfb97
Update inception_v3.py 2021-05-27 14:42:45 +08:00
Bin Lu 06b83fd811
Update mobilenet_v1.py 2021-05-27 14:00:39 +08:00
Bin Lu 7b1dedb807
Update mobilenet_v1.py 2021-05-27 13:50:12 +08:00
Bin Lu 2d1a803964
Update mobilenet_v1.py 2021-05-27 13:36:33 +08:00
Bin Lu 8644331f07
add mobilenet_v1.py to legendary models 2021-05-27 12:10:37 +08:00
Felix d9eeb2d1bb
Update inception_v3.py
remove names and add nn.LayerList
2021-05-26 19:43:27 +08:00
weishengyu 3fa3e577ca remove names 2021-05-26 16:51:42 +08:00
weishengyu ad9caea2db dbg 2021-05-26 16:42:59 +08:00
weishengyu 7eda75225b modify code 2021-05-26 15:47:17 +08:00
Felix a62d5503fc
Create inception_v3.py
add inception_v3
2021-05-26 15:22:01 +08:00
weishengyu c7608cbf20 remove add_sublayers 2021-05-26 14:58:09 +08:00
weishengyu c9d5694cf2 dbg 2021-05-26 14:38:40 +08:00
weishengyu 1e938f7256 remove weight name and add_sublayer 2021-05-26 14:29:56 +08:00
littletomatodonkey 83056d44d5
add vgg network (#743)
* add vgg network

* fix base class

* fix relu and flatten
2021-05-26 11:00:54 +08:00
weishengyu 1d39358358 dbg 2021-05-26 00:31:08 +08:00
weishengyu 0e568006a8 dbg 2021-05-26 00:17:38 +08:00
weishengyu 64345cc0b1 dbg 2021-05-26 00:15:43 +08:00
weishengyu f645fac12d remove Layer1 2021-05-26 00:02:11 +08:00
weishengyu 15884fe78f remove layer1 2021-05-25 23:58:51 +08:00
weishengyu 94b1bae6ae dbg 2021-05-25 23:55:12 +08:00
weishengyu 1153ad6538 dbg 2021-05-25 23:54:23 +08:00
weishengyu 296294b91f dbg 2021-05-25 23:46:11 +08:00
weishengyu a0054caad4 dbg 2021-05-25 23:44:11 +08:00
weishengyu a2367a92b9 dbg 2021-05-25 23:43:41 +08:00
weishengyu 4d4557a32d dbg 2021-05-25 23:38:59 +08:00
weishengyu 9ab8639c3d remove transition 2021-05-25 23:37:43 +08:00
cuicheng01 465a8e5fd3
Create resnet.py 2021-05-25 19:18:14 +08:00
weishengyu b38b0f388f dbg 2021-05-25 17:23:17 +08:00
weishengyu be83d6ae6f use dict instead of list 2021-05-25 17:21:07 +08:00
weishengyu 4e5b60c082 remove name of ConvBN 2021-05-25 17:19:33 +08:00
weishengyu 9b61df62fd replace add_sublayer with nn.Sequential 2021-05-25 17:16:44 +08:00
weishengyu a16de6da0f remove name of conv and bn 2021-05-25 16:55:38 +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