J-shang
|
27a24a12af
|
[Compression] merge nni.contrib.compression with nni.compression (#5573)
Co-authored-by: nishang <nishang@microsoft.com>
|
2023-07-10 10:33:53 +08:00 |
super-dainiu
|
8dc1a83047
|
[compression] fix mask conflict v2 (#5592)
|
2023-07-03 12:10:55 +08:00 |
Hzbeta
|
60c9459205
|
[BugFix] Using the builtin types like `int`, `bool` and `float`. (#5620)
|
2023-06-29 16:37:59 +08:00 |
Bonytu
|
9053d651a6
|
[Compression] Add support for deepspeed (#5517)
Co-authored-by: xinzhang3 <xinzhang3@microsoft.com>
|
2023-06-28 16:43:42 +08:00 |
super-dainiu
|
a9ff00b7e5
|
[compression] pruning dependency v2 (#5569)
|
2023-06-26 15:02:29 +08:00 |
Juntao Liang
|
c55547220f
|
[Common] fix dce and add Scope (#5590)
|
2023-06-01 18:07:32 +08:00 |
super-dainiu
|
aec1fec127
|
[common] cpu/gpu mix trace fix (#5583)
|
2023-05-29 21:46:36 +08:00 |
Yuge Zhang
|
277cf6a19e
|
Remove Retiarii (#5580)
|
2023-05-29 10:46:05 +08:00 |
Yuge Zhang
|
a58cadda48
|
[NAS] Support of PyTorch (and Lightning) 2.0 (#5466)
|
2023-05-29 10:45:42 +08:00 |
Avi Singhal
|
1735a23ffe
|
add avgpool2_formula to shape_formula.py (#5565)
|
2023-05-20 15:27:49 +08:00 |
Super Daniel
|
eb96043c8a
|
[Hotfix] update mmdet to 3.0 (#5515)
|
2023-04-12 10:27:46 +08:00 |
Bonytu
|
b2e2a4d840
|
[Compression] Add support for torch 2.0 (#5492)
|
2023-04-07 13:04:08 +08:00 |
Yihui He
|
3f67d92b67
|
add AMC MIT license (#5498)
|
2023-04-03 10:47:04 +08:00 |
Bonytu
|
32d47768a5
|
[Bugbash] Bug fix (#5467)
|
2023-03-30 13:13:11 +08:00 |
Super Daniel
|
ce6b2e8fc9
|
[Common] Fix some discovered issues during test. (#5487)
|
2023-03-30 10:34:51 +08:00 |
J-shang
|
dd418617b6
|
[Compression] example & test (#5429)
|
2023-03-29 18:35:57 +08:00 |
liuzhe-lz
|
cf5fabd968
|
Merge command channel APIs and add unit tests (#5450)
|
2023-03-20 16:04:00 +08:00 |
Yuge Zhang
|
1f6aedc48f
|
NAS benchmark (stage 2) - space and evaluator (#5380)
|
2023-03-20 10:45:45 +08:00 |
Yuge Zhang
|
13028280ae
|
Enable and fix pipeline issues in NAS (#5439)
|
2023-03-16 17:15:02 +08:00 |
J-shang
|
f8d85ce352
|
[Compression] fix tests & bugs (#5444)
|
2023-03-16 01:32:48 +08:00 |
liuzhe-lz
|
d23dec38e2
|
Remote training service v3 (#5428)
|
2023-03-16 01:32:19 +08:00 |
Louis-J
|
4b967f1ed1
|
feat(speedup): support to call "super()" with concrete trace util. (#5432)
|
2023-03-15 10:39:36 +08:00 |
Bonytu
|
35de06e3f4
|
[Compression] Quantization: add quantizers (#5401)
Co-authored-by: J-shang <shangning128@163.com>
|
2023-03-14 15:51:48 +08:00 |
Yuge Zhang
|
55d7ffb133
|
NAS benchmark (stage 1) - move files (#5379)
|
2023-03-10 12:42:42 +08:00 |
Yuge Zhang
|
7a25f7d78d
|
NAS experiment (stage 3) - main body (#5378)
|
2023-03-10 12:42:32 +08:00 |
junesnow0806
|
8f927a85e4
|
[Compression] modify concrete_trace_utils for producing fx graph (#5403)
|
2023-03-09 17:07:16 +08:00 |
Bonytu
|
f875f47a8b
|
[Compression] Quantization: add module fusion (#5400)
|
2023-03-09 17:06:33 +08:00 |
J-shang
|
7a8fa039ff
|
[Compression] add basic distiller (#5397)
|
2023-03-08 10:34:38 +08:00 |
Yuge Zhang
|
796ee9cf8a
|
NAS experiment (stage 2) - config (#5377)
|
2023-03-08 10:33:34 +08:00 |
J-shang
|
830ec1a807
|
[Compression] pruning stage 4: update speedup support for embedding (#5390)
|
2023-03-07 17:14:08 +08:00 |
Yuge Zhang
|
1c6f61bf97
|
NAS experiment (stage 1) - checkpoint serialization utils (#5376)
|
2023-03-07 13:07:53 +08:00 |
Yuge Zhang
|
5e2c528179
|
NAS strategy (stage 4) - Evolution, HPO and oneshot strategy (#5374)
|
2023-03-07 13:07:37 +08:00 |
Yuge Zhang
|
33d41faac4
|
NAS strategy (stage 5) - middleware (#5375)
|
2023-03-04 00:45:03 +08:00 |
J-shang
|
6bd93e5ad1
|
[Compression] pruning speedup: support input/output masks (#5385)
|
2023-03-03 22:15:44 +08:00 |
J-shang
|
e7828fa32b
|
[Compression] pruning stage 0: update base class & pruning utils (#5386)
|
2023-03-03 21:29:53 +08:00 |
Yuge Zhang
|
d40f408a0a
|
NAS oneshot (stage 2) - Supernet modules (#5372)
|
2023-03-03 15:33:32 +08:00 |
Yuge Zhang
|
4bd3f33a3a
|
NAS oneshot (stage 5) - strategy (#5370)
|
2023-03-03 13:47:57 +08:00 |
Yuge Zhang
|
1cfab22bcd
|
NAS oneshot (stage 4) - Fix issues with ProxylessNAS (#5369)
|
2023-03-03 13:42:10 +08:00 |
Yuge Zhang
|
b032091bcc
|
NAS strategy (stage 3) - RL strategy (#5368)
|
2023-03-03 13:40:42 +08:00 |
Yuge Zhang
|
4b6c0be045
|
NAS strategy (stage 2) - utils and bruteforce strategy (#5367)
|
2023-03-03 09:49:45 +08:00 |
Yuge Zhang
|
808ac9a3fc
|
NAS profiler (stage 3) - FLOPs and nn-Meter (#5363)
|
2023-03-02 20:20:05 +08:00 |
Yuge Zhang
|
e94a81edd0
|
NAS oneshot (stage 1) - Expression and profiler utils (#5366)
|
2023-03-02 11:05:04 +08:00 |
Yuge Zhang
|
7882c628f1
|
NAS profiler (stage 2) - Shape inference and utils (#5362)
|
2023-03-02 11:04:34 +08:00 |
Yuge Zhang
|
e94eb99ce2
|
Rewrite resume and view for experiment (#5365)
|
2023-03-01 10:44:15 +08:00 |
QuanluZhang
|
293cb17004
|
[Compression] Improve quantization speedup with TensorRT (#5039)
|
2023-02-28 08:55:08 +08:00 |
Yuge Zhang
|
67a61e1d94
|
NAS execution engine (stage 3) - CGO (#5360)
|
2023-02-27 20:23:04 +08:00 |
Yuge Zhang
|
1e4f3f08d3
|
NAS execution engine (stage 2) - Sequential and TS (#5359)
|
2023-02-27 20:19:01 +08:00 |
Yuge Zhang
|
03944bd443
|
Refactor nas.nn (Stage 4) - Model Space Hub (#5282)
|
2023-02-24 10:51:05 +08:00 |
Louis-J
|
a4356c07b7
|
feat(speedup): refactor ModulSpeedup V2 using torch.fx graph (#5250)
Co-authored-by: J-shang <shangning128@163.com>
|
2023-02-20 13:58:56 +08:00 |
Yuge Zhang
|
1aa5c251ac
|
Refactor nas.nn (Stage 3) - Cell and Repeat (#5281)
|
2023-02-17 10:39:05 +08:00 |