Bin Lu
|
52663edf60
|
Update retrieval.py
|
2021-08-25 13:57:24 +08:00 |
dongshuilong
|
5bf74beb46
|
update according comments
|
2021-08-24 07:07:17 +00:00 |
|
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 |
weishengyu
|
b4678fd3fa
|
revert trainer
|
2021-08-10 11:27:56 +08:00 |
|
weishengyu
|
f41b09ef7f
|
dbg
|
2021-08-08 17:28:57 +08:00 |
|
weishengyu
|
ef13f8c640
|
modify code
|
2021-08-08 15:04:26 +08:00 |
|
weishengyu
|
e232a97a84
|
dbg
|
2021-08-08 02:10:17 +08:00 |
|
weishengyu
|
91af66dccd
|
replace self.model() with self.forward
|
2021-08-08 02:05:26 +08:00 |
|
weishengyu
|
6c164c9e2b
|
dbg
|
2021-08-08 02:02:42 +08:00 |
|
weishengyu
|
6ebe7f09ed
|
add return_inter flag
|
2021-08-08 01:52:18 +08:00 |
|
weishengyu
|
9790cc517f
|
add return_dict to trainer
|
2021-08-08 01:49:26 +08:00 |
zhiqiu
|
bb1376db45
|
add flags setting
|
2021-07-16 13:26:03 +00:00 |
|
Walter
|
73004f78f5
|
add fp16 amp training and dali (#993)
* add dygraph amp train
|
2021-07-14 22:59:23 +08:00 |
Wei Shengyu
|
5cd7a05d68
|
Merge pull request #1034 from cuicheng01/develop
support load pretrain from url
|
2021-07-13 17:35:04 +08:00 |
cuicheng01
|
592f5b10f5
|
support batch mix
|
2021-07-13 06:38:01 +00:00 |
littletomatodonkey
|
4af9f51088
|
fix windows training (#1038)
* fix windows training
* fix typo
|
2021-07-13 13:57:24 +08:00 |
|
cuicheng01
|
0c63f64bec
|
Update trainer.py
|
2021-07-12 08:06:05 +00:00 |
|
cuicheng01
|
f7329ddf74
|
support load pretrain from url
|
2021-07-12 07:45:44 +00:00 |
cuicheng01
|
dfb8e26923
|
Merge pull request #990 from cuicheng01/develop
Update vision_transformer.py
|
2021-06-30 14:48:47 +08:00 |
|
littletomatodonkey
|
0189236bb8
|
add support for vdl (#978)
* add support for vdl
* fix vdl eval
|
2021-06-30 10:52:27 +08:00 |
|
cuicheng01
|
01f1314fbd
|
fix googlenet infer
|
2021-06-29 14:35:26 +00:00 |
|
littletomatodonkey
|
89229c6246
|
fix trainer
|
2021-06-29 10:57:05 +00:00 |
|
dongshuilong
|
914e69676e
|
fix reid recall metric bugs
|
2021-06-22 17:08:02 +08:00 |
Aurelius84
|
d5c1700faf
|
support to_static for benchmark
|
2021-06-21 06:14:06 +00:00 |
littletomatodonkey
|
cb54c43292
|
add feature calc log (#864)
|
2021-06-16 21:12:25 +08:00 |
|
littletomatodonkey
|
2e6dfa4433
|
fix logger (#840)
* fix logger
* fix trainer for int64 on windows
|
2021-06-15 14:11:50 +08:00 |
|
Bin Lu
|
3ba4503994
|
Merge branch 'PaddlePaddle:develop_reg' into develop_reg
|
2021-06-12 08:46:00 +08:00 |
|
littletomatodonkey
|
d49e1296d9
|
Update trainer.py
|
2021-06-12 03:54:56 +08:00 |
|
Bin Lu
|
73c54f6b8f
|
Update trainer.py
|
2021-06-11 14:52:09 +08:00 |
|
littletomatodonkey
|
edec759fa2
|
fix reader
|
2021-06-10 09:19:49 +00:00 |
|
cuicheng01
|
fcc1b857d4
|
fix eval_interval bugs
|
2021-06-10 08:40:05 +00:00 |
|
littletomatodonkey
|
3b4f5f4dfc
|
add distillation and fix some apis (#810)
* fix save load and imagenet dataset
* refine trainer
|
2021-06-10 16:30:05 +08:00 |
Felix
|
47039c826f
|
Update trainer.py
|
2021-06-08 19:25:00 +08:00 |
|
littletomatodonkey
|
b9044c270a
|
add standard logger
|
2021-06-08 05:21:53 +00:00 |
|
weishengyu
|
934de96539
|
rename cam -> unique
|
2021-06-05 23:06:33 +08:00 |
|
weishengyu
|
9e4a1045f0
|
dbg
|
2021-06-05 16:17:14 +08:00 |
|
weishengyu
|
81c864b96d
|
dbg
|
2021-06-05 16:12:35 +08:00 |
|
weishengyu
|
cd7f606a5f
|
dbg
|
2021-06-05 16:02:22 +08:00 |
|
weishengyu
|
2a72c57e92
|
dbg remove same image
|
2021-06-05 15:56:41 +08:00 |
|
weishengyu
|
5597830e4c
|
dbg camera
|
2021-06-05 15:35:20 +08:00 |
|
weishengyu
|
19be351646
|
dbg
|
2021-06-05 15:12:38 +08:00 |
|
weishengyu
|
0a8b1773ba
|
dbg
|
2021-06-05 14:10:20 +08:00 |
|
weishengyu
|
2a0333281f
|
dbg
|
2021-06-05 13:53:41 +08:00 |
|
weishengyu
|
1c7a23c5e7
|
dbg
|
2021-06-05 13:44:58 +08:00 |
|
weishengyu
|
f9b4e0f0bb
|
dbg
|
2021-06-05 11:36:52 +08:00 |
|
weishengyu
|
d954f73888
|
remove to_tensor
|
2021-06-04 23:40:46 +08:00 |
|
weishengyu
|
b48d2c7a97
|
dbg
|
2021-06-04 23:12:07 +08:00 |
|
weishengyu
|
3e4af0448e
|
add default metrics
|
2021-06-04 22:56:12 +08:00 |
|
weishengyu
|
b23e72b17c
|
add combined_metrics
|
2021-06-04 22:27:32 +08:00 |
|
weishengyu
|
b56237dc99
|
update trainer
|
2021-06-04 21:42:23 +08:00 |
|
weishengyu
|
ed098b3c7d
|
modify trainer
|
2021-06-04 21:11:01 +08:00 |
|
dongshuilong
|
3ef9993026
|
fix ReID trainer bugs
|
2021-06-03 20:14:45 +08:00 |
|
dongshuilong
|
c784209355
|
fix ReID trainer
|
2021-06-03 19:24:50 +08:00 |
|
weishengyu
|
69f563d234
|
rename losses -> loss
|
2021-06-03 15:17:49 +08:00 |
|
dongshuilong
|
435144b526
|
fix bugs imgnet and rec train
|
2021-06-03 12:24:48 +08:00 |
|
dongshuilong
|
452f532121
|
add CompCars train and fix bugs
|
2021-06-03 11:00:47 +08:00 |
|
dongshuilong
|
bba0cf8fdb
|
add CompCars train
|
2021-06-02 20:04:24 +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 |
|
littletomatodonkey
|
dd79f81fd7
|
[WIP]add arch init (#744)
* polish trainer
|
2021-05-27 18:41:44 +08:00 |
|
weishengyu
|
f6da3bd60a
|
move files
|
2021-05-24 11:42:24 +08:00 |