gaotingquan
|
683adcda46
|
fix: support AMP infer
|
2022-05-18 18:10:44 +08:00 |
|
gaotingquan
|
5f88903e6e
|
fix: -> DistributedBatchSampler
|
2022-05-18 18:10:44 +08:00 |
cuicheng01
|
03142ea32b
|
Merge pull request #1919 from cuicheng01/develop
Add PPHGNet code
|
2022-05-13 17:59:53 +08:00 |
cuicheng01
|
8a760fb85f
|
Add PPHGNet code
|
2022-05-12 15:53:51 +00:00 |
|
gaotingquan
|
6e0900ecec
|
feat: add PPLCNetV2_base config
|
2022-05-12 14:38:17 +00:00 |
WangChen0902
|
283ae9b327
|
add dkd (#1888)
* add dkd
* update dkd
* update dkd
* update dkd
* update dkd
* update dkd
* update dkd and add tipc
|
2022-05-09 22:55:01 +08:00 |
littletomatodonkey
|
bb13f3c4f5
|
fix single card dist (#1889)
* fix single card logit
* fix distillation yaml files
|
2022-05-05 09:48:56 +08:00 |
|
gaotingquan
|
24372cc6e5
|
update: update the default gpu num to 8 when using AdamW
|
2022-04-29 17:31:12 +08:00 |
Wei Shengyu
|
e6feb68bb8
|
Merge pull request #1824 from TingquanGao/dev/spt_amp_eval
fix: fp32 eval by default when enable amp
|
2022-04-20 14:40:46 +08:00 |
flytocc
|
c3018ebd39
|
add VAN-Classification code
|
2022-04-19 13:13:33 +08:00 |
|
gaotingquan
|
83ed5195c3
|
fix: set use_fp16_test to True when AMP O2 is enabled
|
2022-04-18 06:14:43 +00:00 |
|
gaotingquan
|
efde56ffc6
|
fix: only fp16 evaluation is supported when ampO2 is enabled
|
2022-04-13 12:14:14 +00:00 |
|
cuicheng01
|
44b9963b25
|
add mobilevit code
|
2022-04-01 08:51:42 +00:00 |
|
cuicheng01
|
f977c7c060
|
update cswin config
|
2022-03-30 03:40:27 +00:00 |
|
cuicheng01
|
51e06d43f5
|
add cswin code and doc
|
2022-03-29 09:50:44 +00:00 |
|
gaotingquan
|
456f9eac81
|
rename amp_O2 to amp_O2_ultra
|
2022-03-11 03:02:07 +00:00 |
|
gaotingquan
|
218da4ddf9
|
feat: amp_O1 -> amp_O1_ultra, amp_O2 -> amp_O2_ultra
|
2022-03-10 11:10:37 +08:00 |
|
WangChen0902
|
7595ba6d70
|
add AFD (#1683)
* add AFD
|
2022-02-28 19:11:50 +08:00 |
|
littletomatodonkey
|
b31b07de45
|
improve distillation config
|
2022-02-14 16:34:01 +08:00 |
|
Tingquan Gao
|
0d22017e2e
|
fix: move class_num from Global to Arch
|
2022-01-25 15:06:36 +08:00 |
|
gaotingquan
|
10c93c55d1
|
fix: enable amp only in training
|
2022-01-25 11:58:07 +08:00 |
|
gaotingquan
|
7040ce8314
|
refactor: change params to be consistent with amp
|
2022-01-25 11:58:07 +08:00 |
|
Wei Shengyu
|
d6d5efe055
|
Merge pull request #1616 from JamesLim-sy/change_some_settings_for_clas_train_performance
Change some settings for clas model`s trainning performance
|
2022-01-25 11:29:27 +08:00 |
|
Tingquan Gao
|
ce166b6ced
|
docs: fix invalid links
|
2022-01-17 21:14:38 +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
|
7a0ed6f08d
|
rename
|
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 |
|
Wei Shengyu
|
0f35f706b6
|
Fix static training speed (#1590)
* fix training speed
* update config setting method
|
2021-12-23 11:13:51 +08:00 |
|
cuicheng01
|
897760f073
|
Merge pull request #1544 from cuicheng01/update_vit
update vit
|
2021-12-14 14:40:14 +08:00 |
|
cuicheng01
|
f5450c3d66
|
delete vit_huge config
|
2021-12-14 03:10:07 +00:00 |
dongshuilong
|
ab187e21cc
|
fix tipc inference batch_size bug
|
2021-12-10 08:05:36 +00:00 |
|
cuicheng01
|
71abe2e9bf
|
update PP-LCNet config
|
2021-12-01 04:08:38 +00:00 |
|
cuicheng01
|
423144c097
|
fix some configs
|
2021-11-15 07:26:58 +00:00 |
|
cuicheng01
|
d868a63764
|
Add ESNet code and pretrained models
|
2021-11-11 06:00:32 +00:00 |
|
gaotingquan
|
720c451bc4
|
fix: fix batch size to 64 * 4
|
2021-11-10 00:16:23 +08:00 |
|
gaotingquan
|
ba2dd01a13
|
refactor: deprecate MixCELoss
|
2021-10-27 19:47:43 +08:00 |
|
cuicheng01
|
ab92b397e0
|
Merge pull request #1300 from TingquanGao/dev/fix_config
fix: fix config
|
2021-10-15 15:37:19 +08:00 |
|
gaotingquan
|
8a5624f835
|
fix: fix config
|
2021-10-12 09:05:03 +00:00 |
|
littletomatodonkey
|
f8b420946c
|
fix cspnet (#1282)
* fix cspnet
* minor fix
|
2021-10-11 17:23:31 +08:00 |
|
gaotingquan
|
283f9fa9b6
|
fix: fix swin training config
|
2021-09-18 12:44:51 +08:00 |
|
gaotingquan
|
f5b32a02ea
|
fix: fix the training configs of deit, swin, twins
|
2021-09-17 20:17:38 +08:00 |
|
cuicheng01
|
48494ec051
|
Merge pull request #1213 from cuicheng01/develop
Add PP-LCNet codes and configs
|
2021-09-17 15:29:50 +08:00 |
|
cuicheng01
|
d7181f6451
|
Merge pull request #1241 from cuicheng01/fix_Alexnet_lr_bug
fix Alexnet lr bug
|
2021-09-17 15:26:47 +08:00 |
|
cuicheng01
|
43aeb572a3
|
fix Alexnet lr bug
|
2021-09-16 08:08:41 +00:00 |
|
cuicheng01
|
4d1f61c407
|
fix Alexnet lr bug
|
2021-09-16 07:54:16 +00:00 |