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 |
|