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