Commit Graph

117 Commits (c30df630356604fe0846de769d92a04d0130af61)

Author SHA1 Message Date
gaotingquan 9beb154bc3 support ShiTu 2023-03-10 16:56:55 +08:00
gaotingquan 32593b6375 refactor 2023-03-10 16:56:55 +08:00
gaotingquan d3941dc1e9 rm amp code from train and eval & use decorator for amp training 2023-03-10 16:56:55 +08:00
gaotingquan 5a6fe171a7 refactor: rm train and eval from engine 2023-03-10 16:56:55 +08:00
gaotingquan 73e2cde617 mv some attrs to __init__() 2023-03-10 16:56:55 +08:00
gaotingquan 0d7e595fc7 mv model_saver to __init__() 2023-03-10 16:56:55 +08:00
gaotingquan 6e77bd6cd5 rm codes for compatibility with old version 2023-03-10 16:56:55 +08:00
gaotingquan f525cea006 replace the arg engine with config 2023-03-10 16:56:55 +08:00
gaotingquan 0e28a39da3 refactor 2023-03-10 16:56:55 +08:00
gaotingquan fad5c8e348 refactor: simpfy engine.train()
1. ModelSaver();
2. _build_ema_model();
3. _init_checkpoints();
4. others.
2023-03-10 16:56:55 +08:00
gaotingquan a38e42f644 refactor: iter_per_epoch -> max_iter 2023-03-10 16:56:55 +08:00
gaotingquan 284e2a6756 refactor: mv all dataloaders to engine.dataloader_dict 2023-03-10 16:56:55 +08:00
gaotingquan 6bed0f5707 refactor: build_train_func & build_eval_func
1. rm engine.device and use paddle.device.get_device() instead;
2. mv some code to build_train_func or build_eval_func to simpfy engine.
2023-03-10 16:56:55 +08:00
gaotingquan 75a20ba557 refactor: add ClassModel to unify model forward interface 2023-03-10 16:56:55 +08:00
gaotingquan 376d83d46e refactor: simplify engine 2023-03-10 16:56:55 +08:00
gaotingquan 81de331e00 rename to re_parameterize() for re-parameterization nets 2023-01-06 20:29:07 +08:00
HydrogenSulfate 97398d9f12 fix export pretrained_model when it is None 2022-12-16 18:46:21 +08:00
HydrogenSulfate 51d71016ac support web link for 'Global.pretrained_model' when export models 2022-12-16 18:46:21 +08:00
HydrogenSulfate 6cc6540ca5 add different seed for workers and replicas 2022-11-24 11:51:13 +08:00
HydrogenSulfate 1ad4167fdc fix engine.py 2022-11-22 16:47:47 +08:00
HydrogenSulfate 25ed25372a fix engine.py 2022-11-22 16:47:47 +08:00
HydrogenSulfate c07758b331 fix engine.py 2022-11-22 16:47:47 +08:00
HydrogenSulfate 7ff257ea69 fix random seed bug for pksampler in DDP 2022-11-22 16:47:47 +08:00
USTCKAY 0cec70bd22 [CustomDevice]add support for custom NPU, test=develop 2022-10-26 10:33:10 +08:00
dongshuilong 944763d7a5 add fixmatch 2022-10-25 12:04:22 +08:00
HydrogenSulfate 09200a31f4 remove redundant code, fix bugs in lr.step, merge GoodsDataset into Vehicle 2022-10-19 07:18:21 +00:00
HydrogenSulfate 2cfd8dd828 add xbm 2022-10-17 15:45:45 +08:00
cuicheng01 0b9f62262a
Merge branch 'develop' into fix_multilabel 2022-09-14 14:41:32 +08:00
HydrogenSulfate 5a4874079d fix some bugs 2022-08-23 10:59:30 +00:00
HydrogenSulfate 1b5e00e82a add PP-ShiTuV2 code 2022-08-23 07:54:58 +00:00
cuicheng01 eade1b7232 fix multilabel 2022-08-22 06:40:26 +00:00
Yang Nie 436f64ad84 Merge branch 'develop' into ConvNeXt 2022-07-09 21:50:31 +08:00
HydrogenSulfate e8fc16d76b update rep condition 2022-06-28 05:58:14 +00:00
HydrogenSulfate de1e9cab74 update pact chain 2022-06-28 04:06:54 +00:00
littletomatodonkey 794af8c06f
add kd doc (#1997)
* add kd doc

* fix

* add ssld doc

* fix ssld

* fix ssld

* Update knowledge_distillation.md

* fix doc

* fix dist export

* fix

* add dist doc

* fix speed info

* Update ssld.md

* Update ssld.md
2022-06-09 14:52:50 +08:00
Tingquan Gao b457c393eb
fix: dbg (#1993)
1. config["Metric"]["Train"] may be None
2. fliter -> list
2022-06-07 20:08:48 +08:00
gaotingquan 80358efbe2 fix: pop TopkAcc when using batch_transform_ops 2022-06-06 16:47:57 +08:00
gaotingquan ccefbd7864 fix: dbg 2022-06-06 16:47:57 +08:00
gaotingquan df3e75dde4 fix: warn when topk parameter setting is wrong 2022-06-06 16:47:57 +08:00
littletomatodonkey 4003cdb7ad
add traffic sign yaml (#1966)
* add traffic sign and vehicle attr doc and yaml
2022-06-06 13:13:50 +08:00
Wei Shengyu d039691fce
Merge pull request #1963 from TingquanGao/dev/rm_fluid
remove fluid
2022-06-06 10:54:16 +08:00
cuicheng01 ad71254e68
Merge pull request #1960 from zhiboniu/dev_attr
support attribute infer
2022-05-27 19:22:17 +08:00
Walter dcef409dde
Merge branch 'develop' into adaface 2022-05-27 14:19:03 +08:00
gaotingquan c22bdc7e54
remove fluid 2022-05-26 07:40:15 +00:00
zhiboniu a4e1da6610 modify attr export model 2022-05-25 08:13:38 +00:00
littletomatodonkey 6dac07f68d
fix quant logic (#1941)
* fix quant logic

* add support for trt+int8 inference

* add support for swin quant

* fix swin and quant

* fix assert info

* fix assert info

* fix log
2022-05-23 19:04:27 +08:00
cuicheng01 8f5fe61540
Merge pull request #1925 from cuicheng01/add_person_demo
Add person demo
2022-05-23 09:26:49 +08:00
gaotingquan 683adcda46 fix: support AMP infer 2022-05-18 18:10:44 +08:00
gaotingquan e4327c8686 log tip when export succeeded 2022-05-18 13:00:00 +08:00
littletomatodonkey afafb8f41d fix save load 2022-05-17 13:24:24 +00:00