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 |
|
littletomatodonkey
|
aea712cc87
|
add dist of rec model (#1574)
* add distillation loss func and rec distillation
|
2022-01-05 19:25:36 +08: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 |
|
gaotingquan
|
5d53e9f152
|
fix: raise warning when setting batch_transform_ops and TopkAcc
|
2021-12-24 21:22:52 +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 |
|
cuicheng01
|
897760f073
|
Merge pull request #1544 from cuicheng01/update_vit
update vit
|
2021-12-14 14:40:14 +08:00 |
|
cuicheng01
|
b3922c96f5
|
Merge pull request #1436 from TingquanGao/dev/update_whl
update whl
|
2021-12-14 12:15:46 +08:00 |
|
cuicheng01
|
f5450c3d66
|
delete vit_huge config
|
2021-12-14 03:10:07 +00:00 |
|
cuicheng01
|
83d1b411c5
|
update vit
|
2021-12-10 09:56:33 +00:00 |
Walter
|
524190a775
|
Merge pull request #1543 from RainFrost1/tipc_batch_size
fix tipc batch_size bug
|
2021-12-10 16:57:36 +08:00 |
|
dongshuilong
|
ab187e21cc
|
fix tipc inference batch_size bug
|
2021-12-10 08:05:36 +00:00 |
|
Wei Shengyu
|
51c3a34275
|
Merge pull request #1540 from weisy11/refor_quant
Refor quant
|
2021-12-10 14:59:39 +08:00 |
weishengyu
|
534037d145
|
dbg
|
2021-12-10 11:14:14 +08:00 |
|
weishengyu
|
6510da8fca
|
dbg
|
2021-12-09 20:15:25 +08:00 |
|
weishengyu
|
ad1a2fd137
|
move slim into arch
|
2021-12-09 20:08:57 +08:00 |