Tingquan Gao
|
6aabb94d8c
|
Revert "refactor: add ClassModel to unify model forward interface"
This reverts commit 75a20ba557.
|
2023-03-14 16:47:13 +08:00 |
|
Tingquan Gao
|
502aead98b
|
Revert "backbone_zoo -> backbone"
This reverts commit e4a3e1bb12.
|
2023-03-14 16:47:13 +08:00 |
|
Tingquan Gao
|
f91811dab9
|
Revert "use decorator to parse batch"
This reverts commit 97935164fe.
|
2023-03-14 16:47:13 +08:00 |
|
Tingquan Gao
|
aa52682c55
|
Revert "rm amp code from train and eval & use decorator for amp training"
This reverts commit d3941dc1e9.
|
2023-03-14 16:47:13 +08:00 |
|
Tingquan Gao
|
876732fbcc
|
Revert "debug"
This reverts commit ac27cb1917.
|
2023-03-14 16:47:13 +08:00 |
gaotingquan
|
ac27cb1917
|
debug
|
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
|
97935164fe
|
use decorator to parse batch
|
2023-03-10 16:56:55 +08:00 |
|
gaotingquan
|
e4a3e1bb12
|
backbone_zoo -> backbone
|
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 |
zh-hike
|
d8f049ae76
|
增加代码规范，删除无用空格
|
2023-01-04 13:42:00 +08:00 |
|
zh-hike
|
c5a29dbaec
|
代码规范
|
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 |
|
gaotingquan
|
c1a8ce4c94
|
fix error that sync bn should not be used on cpu
|
2022-11-16 16:05:53 +08:00 |
root
|
2a909c306a
|
fix: change to relative import
|
2022-08-19 11:40:38 +08:00 |
|
gaotingquan
|
20730fad07
|
revert
the code may cause problem that loss change in training.
|
2022-05-24 22:19:10 +08: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 |
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 |