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 |
xiongkun
|
821e550900
|
stop_gradient=True for input spec (#1943)
|
2022-05-23 09:48: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 |
cuicheng01
|
8b639e939f
|
update code and docs
|
2022-05-23 01:11:03 +00:00 |
gaotingquan
|
675e60d5a5
|
feat: log the FLOPs and Params when building model
|
2022-05-17 21:24:17 +08:00 |
|
cuicheng01
|
d125b31f73
|
update code
|
2022-05-14 10:03:55 +00:00 |
WangChen0902
|
7595ba6d70
|
add AFD (#1683)
* add AFD
|
2022-02-28 19:11:50 +08:00 |
|
littletomatodonkey
|
e0a6e5bf38
|
fox dist err (#1621)
* fox dist err
* fix init
* fix init
|
2022-01-06 09:46:52 +08:00 |
|
littletomatodonkey
|
aea712cc87
|
add dist of rec model (#1574)
* add distillation loss func and rec distillation
|
2022-01-05 19:25:36 +08:00 |
weishengyu
|
ad1a2fd137
|
move slim into arch
|
2021-12-09 20:08:57 +08:00 |
|
weishengyu
|
6c5d1ebc28
|
add pruner and quanter for theseus
|
2021-12-09 14:51:40 +08:00 |
|
weishengyu
|
600067f4fc
|
dbg
|
2021-09-15 11:35:49 +08:00 |
|
weishengyu
|
e916259508
|
update return_res method
|
2021-09-15 11:03:46 +08:00 |
Aurelius84
|
d5c1700faf
|
support to_static for benchmark
|
2021-06-21 06:14:06 +00:00 |
|
littletomatodonkey
|
35437e2236
|
fix distillation model infer and export model
|
2021-06-13 05:01:56 +00:00 |
|
littletomatodonkey
|
edec759fa2
|
fix reader
|
2021-06-10 09:19:49 +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 |
|
weishengyu
|
91f72dcef7
|
dbg label
|
2021-06-05 17:56:40 +08:00 |
|
weishengyu
|
ce43150ffe
|
dbg
|
2021-06-05 17:45:31 +08:00 |
dongshuilong
|
8a8eb34d79
|
add vehicle neck and fix bugs
|
2021-06-04 16:44:24 +08:00 |
|
dongshuilong
|
505c9309a0
|
add vehicle neck
|
2021-06-04 14:54:34 +08:00 |
|
weishengyu
|
586af751ac
|
rename head -> gears
|
2021-06-03 15:19:39 +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 |
Bin Lu
|
85ca658ea2
|
Update __init__.py
|
2021-06-01 11:30:26 +08:00 |
|
littletomatodonkey
|
dd79f81fd7
|
[WIP]add arch init (#744)
* polish trainer
|
2021-05-27 18:41:44 +08:00 |
|
weishengyu
|
611a40958a
|
dbg
|
2021-05-25 16:40:20 +08:00 |
|
weishengyu
|
f6da3bd60a
|
move files
|
2021-05-24 11:42:24 +08:00 |