Граф коммитов

475 Коммитов

Автор SHA1 Сообщение Дата
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