6cb17cfe44
fix: change to relative import
2022-08-23 17:24:34 +08:00
7595ba6d70
add AFD ( #1683 )
...
* add AFD
2022-02-28 19:11:50 +08:00
6963c25804
fix: fix bug
...
bug is that error rasied by calling update_res() when sublayer does not have _save_sub_res_hook() because nn.Sequential and nn.LayerList is not TheseusLayer.
2022-01-14 14:58:28 +08:00
43fce425ac
fix: change the returned result from dict to list
2022-01-14 14:58:28 +08:00
3b9f629220
feat: support specify return_stages to update_res()
2022-01-14 14:58:28 +08:00
71cb728b07
fix: rename to upgrade_sublayer()
2021-12-28 15:24:50 +08:00
a86c4b290e
add license
2021-12-28 15:24:50 +08:00
721ac0bf61
refactor: simplify code
...
1. remove WrapLayer and wrap_theseus;
2. support call update_res() one more;
3. optim parse_pattern_str() to return list of layer parsed.
2021-12-28 15:24:50 +08:00
56911b573b
refactor: rename replace_sub() func to ?
2021-12-28 15:24:50 +08:00
0f126b75da
refactor: strengthen parse_pattern_str() func
2021-12-28 15:24:50 +08:00
8d0b0d4b0a
refactor: extract _parse_pattern_str() func
2021-12-28 15:24:50 +08:00
18dec0744a
fix: fix problems commented in reviewing
2021-12-28 15:24:50 +08:00
41296972a4
fix: fix comments
2021-12-28 15:24:50 +08:00
f8ee6c0f86
fix: fix result returned by stop_after
2021-12-28 15:24:50 +08:00
cf205e1379
fix: fix result returned by _find_layers_handle
2021-12-28 15:24:50 +08:00
b0ae3a1210
fix: unify the pattern of layer's name
2021-12-28 15:24:50 +08:00
6c5d1ebc28
add pruner and quanter for theseus
2021-12-09 14:51:40 +08:00
3c148cb56a
update return res method and fix amp bug ( #1309 )
...
* fix amp bug; update return res method
* support index for list object for return res
2021-10-15 18:25:50 +08:00
e916259508
update return_res method
2021-09-15 11:03:46 +08:00
5131956d15
dbg theseus
2021-09-06 16:18:57 +08:00
98f38fa0f7
dbg
2021-08-08 17:53:06 +08:00
f41b09ef7f
dbg
2021-08-08 17:28:57 +08:00
88a72b6735
dbg
2021-08-08 15:56:57 +08:00
2cb3bf66dd
update wrap theseus rule
2021-08-08 15:55:34 +08:00
1985075550
dbg
2021-08-08 15:18:28 +08:00
ef13f8c640
modify code
2021-08-08 15:04:26 +08:00
6dbbf8cc17
add theseus wrapper
2021-08-08 14:57:29 +08:00
4f8552aa4f
dbg
2021-08-08 14:32:45 +08:00
132ec21fba
reverse res_dict
2021-08-08 14:27:43 +08:00
56c513fdc7
remove disconnect
2021-08-08 13:29:30 +08:00
7e1aa2d358
dbg
2021-08-08 12:58:31 +08:00
6ebe7f09ed
add return_inter flag
2021-08-08 01:52:18 +08:00
9790cc517f
add return_dict to trainer
2021-08-08 01:49:26 +08:00
bba9e0dfff
add vgg theseus funcs
2021-08-08 01:07:49 +08:00
8c1515b5db
add hook
2021-08-08 00:37:53 +08:00
0040a973eb
add save res back
2021-08-07 23:00:32 +08:00
e11cc5c8c1
remove useless comments
2021-06-16 22:13:32 +08:00
76fcde59af
add variant model example
2021-06-04 10:19:01 +08:00
2b150a2ffa
dbg
2021-05-25 16:51:04 +08:00
88f92297fa
dbg
2021-05-25 16:50:37 +08:00
79e9419fad
dbg
2021-05-25 16:47:33 +08:00
608b163d4f
move stop_after out of update_sub, and rename update_sub to update_res
2021-05-24 19:49:06 +08:00
f6da3bd60a
move files
2021-05-24 11:42:24 +08:00
root
WangChen0902
gaotingquan
weishengyu
Wei Shengyu
dongshuilong