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 |
tianyi1997
|
f313a6d873
|
Support training without amp
|
2023-02-28 15:01:21 +08:00 |
|
tianyi1997
|
a9b8432597
|
fix: wrong base class & simplify train func
|
2023-02-28 15:01:21 +08:00 |
|
tianyi1997
|
ac466b68da
|
Create MetaBNNeck
|
2023-02-28 15:01:21 +08:00 |
|
tianyi1997
|
21e76d08b6
|
Modify codes based on reviews
|
2023-02-28 15:01:21 +08:00 |
|
tianyi1997
|
d79fb66e19
|
Create train func & cfg for MetaBIN
|
2023-02-28 15:01:21 +08:00 |
HydrogenSulfate
|
fb763b7096
|
fix(dali): reset dataloader manually when using DALI dataloader during epoch training
|
2023-02-21 20:11:04 +08:00 |
zengshao0622
|
1c68b63a66
|
fix some nots
|
2023-02-15 12:00:55 +08:00 |
|
HydrogenSulfate
|
4fdcda7c60
|
fix bug in randaug, train_progressive and efficientnet_v2
|
2023-01-31 21:15:22 +08:00 |
|
HydrogenSulfate
|
7e12c73e0a
|
polish progressive training code
|
2023-01-31 21:15:22 +08:00 |
|
HydrogenSulfate
|
3a8b56801d
|
feat(model): add EfficientNetV2 code and fix AttrDict BUG
|
2023-01-31 21:15:22 +08:00 |
|
gaotingquan
|
81de331e00
|
rename to re_parameterize() for re-parameterization nets
|
2023-01-06 20:29:07 +08:00 |
zh-hike
|
4292c1a90d
|
fix turtle import bug
|
2023-01-06 10:57:05 +08:00 |
|
HydrogenSulfate
|
b05c8b1b7e
|
fix(retrieval): remove max_iter in retrieval.py to avoid empty data
|
2023-01-05 18:44:22 +08:00 |
|
HydrogenSulfate
|
2f331d8588
|
use scipy.sparse to speed up local query expansion
|
2023-01-05 15:13:04 +08:00 |
|
HydrogenSulfate
|
6871d601ff
|
speed up local QE by sparse matmul
|
2023-01-05 15:13:04 +08:00 |
|
HydrogenSulfate
|
c6865e255e
|
refactor(retrieval): polish retrieval.py
|
2023-01-05 15:13:04 +08:00 |
|
HydrogenSulfate
|
97f99cd826
|
refactor(retrieval): polish retrieval.py
|
2023-01-05 15:13:04 +08:00 |
|
zh-hike
|
f8978a2f0c
|
修改cifar100参数配置
|
2023-01-04 13:42:00 +08:00 |
|
zh-hike
|
d8f049ae76
|
增加代码规范，删除无用空格
|
2023-01-04 13:42:00 +08:00 |
|
zh-hike
|
f073e97d22
|
增加cifar100参数yaml
|
2023-01-04 13:42:00 +08:00 |
|
zh-hike
|
009f347d64
|
修改空格等代码规范
|
2023-01-04 13:42:00 +08:00 |
|
zh-hike
|
692b8d8c1f
|
增加data的代码复用及RecModel的修改，代码已跑通
|
2023-01-04 13:42:00 +08:00 |
|
zh-hike
|
4db132444d
|
增加RecModel配合WideResNet代码以及参数转换成RecModel
|
2023-01-04 13:42:00 +08:00 |
|
zh-hike
|
7823f340b1
|
cifar10_4000跑出论文精度代码
|
2023-01-04 13:42:00 +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 |
|
zengshao0622
|
cddc899f34
|
fix dataloader next
|
2022-11-09 19:01:00 +08:00 |
|
HydrogenSulfate
|
7d9f4dcb59
|
change Tensor.numpy()[0] to float(Tensor) for 0-D tensor case
|
2022-11-01 14:37:11 +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 |
|
HydrogenSulfate
|
61b4153907
|
add batch Tensor collate to simplify dali code in train/eval/retrival code
|
2022-10-13 12:05:50 +08:00 |
cuicheng01
|
50fc7d0eae
|
fix bugs to adapt to the new framework
|
2022-09-20 10:06:20 +00:00 |
|
cuicheng01
|
cc46db1586
|
fix bugs to adapt to the new framework
|
2022-09-19 02:01:31 +00:00 |