1011 Commits

Author SHA1 Message Date
zhangbo9674
cd039a7b37 add save_dtype 2022-01-10 18:19:03 +08:00
zhangbo9674
d437bb0a7e use fp32 to eval 2022-01-10 18:19:03 +08:00
zhangbo9674
bb19c1f7a6 fix eval bug 2022-01-10 18:19:03 +08:00
zhangbo9674
b2956c1b41 refine code 2022-01-10 18:19:03 +08:00
zhangbo9674
205592a3e3 fix amp with distribute bug 2022-01-10 18:19:03 +08:00
gaotingquan
1a9b1782ef fix: fix min_lr & warmup_lr 2022-01-07 11:52:30 +08:00
gaotingquan
56c72a85a1 fix: fix lr 2022-01-07 11:52:30 +08:00
gaotingquan
025feb6de9 fix: fix -1 in dims 2022-01-07 11:52:30 +08:00
gaotingquan
7a0ed6f08d rename 2022-01-07 11:52:30 +08:00
gaotingquan
ee819393ac fix: fix -1 in dims and ignore swapdim in static 2022-01-07 11:52:30 +08:00
gaotingquan
00fb3f7519 feat: support PVTV2 2022-01-07 11:52:30 +08:00
gaotingquan
a53b636829 feat: support PVTV2 2022-01-07 11:52:30 +08:00
littletomatodonkey
e0a6e5bf38
fox dist err (#1621)
* fox dist err

* fix init

* fix init
2022-01-06 09:46:52 +08:00
JamesLim-sy
f015860645 drop useless settings 2022-01-05 13:15:19 +00:00
littletomatodonkey
aea712cc87
add dist of rec model (#1574)
* add distillation loss func and rec distillation
2022-01-05 19:25:36 +08:00
JamesLim-sy
24a51ae064 first 2022-01-04 07:20:31 +00:00
Bin Lu
304bafb735
Merge pull request #1610 from TingquanGao/dev/fix_theseus
fix: upgrade replace_sub() to upgrade_sublayer()
2021-12-31 12:15:18 +08:00
gaotingquan
d6aeba6f56
fix: upgrade replace_sub() to upgrade_sublayer()
supplement to PR #1581
2021-12-30 10:59:20 +00:00
gaotingquan
6e13ff3068 fix: use hasattr() to check if collate_fn is in dataloader
fix bug caused by PR #1596
2021-12-30 16:35:05 +08:00
gaotingquan
cabc7a0f83 fix: replace reg exp with layer name 2021-12-28 15:24:50 +08:00
gaotingquan
71cb728b07 fix: rename to upgrade_sublayer() 2021-12-28 15:24:50 +08:00
gaotingquan
1e696ac22b fix: remove unnecessary register_hook() call & pre-commit 2021-12-28 15:24:50 +08:00
gaotingquan
a86c4b290e add license 2021-12-28 15:24:50 +08:00
gaotingquan
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
gaotingquan
56911b573b refactor: rename replace_sub() func to ? 2021-12-28 15:24:50 +08:00
gaotingquan
0f126b75da refactor: strengthen parse_pattern_str() func 2021-12-28 15:24:50 +08:00
gaotingquan
8d0b0d4b0a refactor: extract _parse_pattern_str() func 2021-12-28 15:24:50 +08:00
gaotingquan
18dec0744a fix: fix problems commented in reviewing 2021-12-28 15:24:50 +08:00
gaotingquan
41296972a4 fix: fix comments 2021-12-28 15:24:50 +08:00
gaotingquan
f8ee6c0f86 fix: fix result returned by stop_after 2021-12-28 15:24:50 +08:00
gaotingquan
cf205e1379 fix: fix result returned by _find_layers_handle 2021-12-28 15:24:50 +08:00
gaotingquan
b0ae3a1210 fix: unify the pattern of layer's name 2021-12-28 15:24:50 +08:00
gaotingquan
7da2a997e9 fix: save latest model every epoch 2021-12-27 22:04:26 +08:00
weishengyu
377a7b4b2f dbg 2021-12-27 20:35:42 +08:00
weishengyu
9a1d35ba3b add license info 2021-12-27 20:16:56 +08:00
weishengyu
87f508e9f9 dbg 2021-12-27 19:37:00 +08:00
gaotingquan
5d53e9f152 fix: raise warning when setting batch_transform_ops and TopkAcc 2021-12-24 21:22:52 +08:00
weishengyu
b662ed34ac dbg 2021-12-24 11:15:27 +08:00
weishengyu
18d99b012b move gallery layer into extractor 2021-12-24 11:06:42 +08:00
weishengyu
25edd1c0d8 refactor code 2021-12-23 20:51:15 +08:00
weishengyu
d5637367a9 add export method 2021-12-23 20:35:10 +08:00
weishengyu
932e0eace1 dbg 2021-12-23 20:29:05 +08:00
Wei Shengyu
0f35f706b6
Fix static training speed (#1590)
* fix training speed

* update config setting method
2021-12-23 11:13:51 +08:00
Wei Shengyu
f45f9ee4d4
Merge pull request #1585 from zhangbo9674/dev/resnet50_optimize
Accelerate dynamic graph amp training
2021-12-21 17:31:25 +08:00
zhangbo9674
558f03d684 refine code 2021-12-21 06:30:13 +00:00
zhangbo9674
28061f537c refine optimizer init logice 2021-12-21 06:28:13 +00:00
cuicheng01
7aa9f1c6ea
Merge pull request #1588 from cuicheng01/add_new_config
add some new configs
2021-12-21 14:03:51 +08:00
cuicheng01
2e578335cd add some new configs 2021-12-20 08:10:39 +00:00
zhangbo9674
b54ee04491 Accelerate dynamic graph amp training 2021-12-20 06:36:56 +00:00
gaotingquan
7732a69f1b fix: fix key error in distillation 2021-12-16 18:21:08 +08:00