Commit Graph

215 Commits (6b813ec6481ad0a52a67d9b4c645bb7b0ed9a373)

Author SHA1 Message Date
dongshuilong 15f6f58139 refactor trainer v2 2021-08-24 03:02:55 +00:00
dongshuilong ebde0e13cb refactor trainer 2021-08-22 15:10:23 +00:00
cuicheng01 4b10174a41
Update export_model.py 2021-06-29 22:39:06 +08:00
cuicheng01 01f1314fbd fix googlenet infer 2021-06-29 14:35:26 +00:00
littletomatodonkey dca0f73820
rm unused dir (#951)
* rm unused dir

* fix wx group
2021-06-22 21:34:05 +08:00
Wei Shengyu 80bbf62a46
merge develop (#859)
merge develop into reg
2021-06-16 18:43:08 +08:00
littletomatodonkey b3108afa19 fix log with export model 2021-06-15 09:27:54 +00:00
littletomatodonkey 2e6dfa4433
fix logger (#840)
* fix logger
* fix trainer for int64 on windows
2021-06-15 14:11:50 +08:00
littletomatodonkey 35437e2236 fix distillation model infer and export model 2021-06-13 05:01:56 +00:00
littletomatodonkey 061d07cd91
refine tool dir (#827)
* refine tool dir
2021-06-11 12:17:56 +08:00
weishengyu 07089dcfe7 move identity_head to gears 2021-06-07 14:22:53 +08:00
weishengyu e8959c31d5 add default value for label 2021-06-07 09:43:47 +08:00
weishengyu 03d32d61e0 dbg 2021-06-06 18:52:12 +08:00
weishengyu a22b8f6b04 remove neck when useless 2021-06-06 18:20:26 +08:00
weishengyu 16718f08c6 Merge branch 'develop_reg' of https://github.com/weisy11/PaddleClas into develop_reg 2021-06-05 18:05:25 +08:00
weishengyu 99298c8a15 change default config 2021-06-05 18:05:08 +08:00
Wei Shengyu 399ac69f30 Merge branch 'PaddlePaddle:develop_reg' into develop_reg 2021-06-05 18:01:07 +08:00
weishengyu a445ff1f28 update export model 2021-06-05 17:36:24 +08:00
Walter 2f1bd9ab39
Merge pull request #790 from weisy11/develop_reg
modify trainer
2021-06-05 17:02:50 +08:00
weishengyu ed098b3c7d modify trainer 2021-06-04 21:11:01 +08:00
littletomatodonkey c356974e14 add inference and its yaml 2021-06-04 09:28:37 +00:00
dongshuilong 10c0fc4b73 fix ReID trainer bugs 2021-06-03 19:28:32 +08:00
cuicheng01 d13cb22378 Add legendary_model's congigs and fix some trainer's bug 2021-06-01 17:24:21 +00:00
littletomatodonkey 0a0e34298b
add support for infer and export (#760)
* add support for infer

* add export model support

* fix yaml

* fix post process name

* fix topk name
2021-06-01 14:37:47 +08:00
littletomatodonkey 3881343484
add support for train and eval (#752)
* add support for train and eval

* rm unsed code

* add support for metric save and load ckp
2021-05-29 13:52:32 +08:00
weishengyu 920a44b0ce move files 2021-05-24 11:43:47 +08:00
Yiqun Liu 0d832a2539
Enable profiler, for both static and dynamic training. (#729)
* Enable profiler for static training.

* Polish the initialize of ProfilerOptions.

* Enable profiler for dynamic mode.
2021-05-21 10:31:43 +08:00
liuyuhui 5170153ee3 merge with upstream/develop 2021-05-08 11:16:41 +00:00
liuyuhui a9bce0409d fix tools/train.py 2021-05-08 08:42:30 +00:00
liuyuhui 55fa094e60 fix one card eval in multicards training 2021-05-08 05:29:12 +00:00
littletomatodonkey e165897c37
fix drop last for training process (#713) 2021-05-06 14:21:54 +08:00
littletomatodonkey 9c0f049603
fix export model eval (#710)
* adapt to net.eval for the framework just contains training flag setting
* fix bug when export swin transformer
2021-05-06 13:55:11 +08:00
Tingquan Gao dd70cb1bb0
Fix HubServing demo (#708) 2021-04-30 16:10:17 +08:00
liuyuhui 14a93e7933
[Kunlun]add multi xpu support for PaddleClas about dygraph (#690)
* add multi xpu support for PaddleClas about dygraph

* add dygraph multi xpu support
2021-04-29 17:08:26 +08:00
littletomatodonkey f17343b017
add support for eval using inference engine (#696) 2021-04-26 15:18:04 +08:00
gaotingquan 64fad361ab Fix the exception of DALI 2021-04-16 10:55:31 +08:00
liuyuhui c3d401b7ea
add multi xpu support for PaddleClas (#678) 2021-04-14 22:31:36 +08:00
littletomatodonkey a7aa14525c
fix repvgg eval (#677)
* fix repvgg eval

* fix dp training

* fix single card train
2021-04-14 02:16:13 +08:00
littletomatodonkey 2e62e2e25e
fix loss reduce from dict to list (#679)
* fix loss reduce from dict to list

* remove note
2021-04-13 23:19:36 +08:00
littletomatodonkey a6e2114e32
add find_unused_parameters param (#668)
* add find_unused_parameters param

* fix default val
2021-04-06 21:29:38 +08:00
Tingquan Gao 4523d4246d
Adapt paddle_inference 2.0.1 API (#667) 2021-04-06 10:52:40 +08:00
yaohai fee32b555a fix small error 2021-03-30 17:14:25 +08:00
yaohai 5fd7085ddf add multilabel feature 2021-03-30 16:02:32 +08:00
Tingquan Gao 8a469799af
support bs>1 (#651)
* support bs>1
2021-03-26 18:52:50 +08:00
Tingquan Gao 8832a3fa0a
Support Visual DL (#650)
* Support Visual DL

* Fix VDL

* Add doc of VDL, test=document_fix

* Add the en doc of VDL, test=document_fix

Co-authored-by: littletomatodonkey <2120160898@bit.edu.cn>
2021-03-23 12:45:01 +08:00
littletomatodonkey 4ba3d47e31
Merge branch 'develop' into cp_fp16_training 2021-03-01 16:18:26 +08:00
Zhang Ting aeccae2128
fix oom for batch_size=208 (#618) 2021-03-01 12:37:17 +08:00
huangxu96 4e43ec6995 new usage of amp training. (#564)
* new usage of amp training.

* change the usage of amp and pure fp16 training.

* modified code as reviews
2021-02-26 09:25:54 +00:00
littletomatodonkey ba17052a54
add eta info (#613)
* add eta info

* rm duplicate desc
2021-02-26 13:40:40 +08:00
littletomatodonkey c8a155635e
fix train and save (#594) 2021-02-01 22:24:25 +08:00