Commit Graph

142 Commits (c35d3fcda6d38ceb31eff88ac93174a20902fc7d)

Author SHA1 Message Date
littletomatodonkey aea712cc87
add dist of rec model (#1574)
* add distillation loss func and rec distillation
2022-01-05 19:25:36 +08:00
gaotingquan 1e696ac22b fix: remove unnecessary register_hook() call & pre-commit 2021-12-28 15:24:50 +08:00
cuicheng01 fb631b38fb update esnet.py 2021-11-11 12:51:58 +00:00
cuicheng01 d868a63764 Add ESNet code and pretrained models 2021-11-11 06:00:32 +00:00
cuicheng01 48494ec051
Merge pull request #1213 from cuicheng01/develop
Add PP-LCNet codes and configs
2021-09-17 15:29:50 +08:00
Wei Shengyu d98b881603
Merge pull request #1208 from weisy11/fix_bug_of_theseus
dbg theseus
2021-09-15 14:04:51 +08:00
weishengyu e916259508 update return_res method 2021-09-15 11:03:46 +08:00
cuicheng01 d470f34c64 Update code and docs 2021-09-13 03:18:37 +00:00
dongshuilong 8e85ef4775 update slim 2021-09-10 03:09:03 +00:00
cuicheng01 4150df912e Add LCNet codes and config 2021-09-07 16:14:40 +00:00
weishengyu 5131956d15 dbg theseus 2021-09-06 16:18:57 +08:00
dongshuilong 68c1b089e8 update slim for new trainer 2021-08-27 09:32:37 +00:00
weishengyu 98f38fa0f7 dbg 2021-08-08 17:53:06 +08:00
weishengyu f41b09ef7f dbg 2021-08-08 17:28:57 +08:00
weishengyu ef13f8c640 modify code 2021-08-08 15:04:26 +08:00
weishengyu 4f8552aa4f dbg 2021-08-08 14:32:45 +08:00
weishengyu 8211880d10 dbg 2021-08-08 14:29:07 +08:00
weishengyu 132ec21fba reverse res_dict 2021-08-08 14:27:43 +08:00
weishengyu 3b62e4d401 dbg 2021-08-08 13:14:30 +08:00
weishengyu 7e1aa2d358 dbg 2021-08-08 12:58:31 +08:00
weishengyu 9790cc517f add return_dict to trainer 2021-08-08 01:49:26 +08:00
weishengyu bba9e0dfff add vgg theseus funcs 2021-08-08 01:07:49 +08:00
littletomatodonkey 9d9cd3719e
add static training (#1037)
* add static training

* fix typo

* add se fp16

* rm note

* fix loader

* fix cfg
2021-07-15 10:30:07 +08:00
weishengyu 546d63d58a dbg 2021-05-31 16:49:25 +08:00
weishengyu c749bdf93f replace reshape with Flatten layer in ResNet 2021-05-31 16:40:33 +08:00
dongshuilong 2fa808517d legendary models v0.1 2021-05-31 16:01:02 +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
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
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 5671f9d9bb
Update resnet.py 2021-05-28 10:50:17 +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 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