microsoft
/
nni
зеркало из https://github.com/microsoft/nni.git
1
0
Форкнуть 0
Код Задачи Пакеты Проекты Релизы Вики Активность
2 988 коммитов 79 Ветки 49 Теги 420 MiB
Граф коммитов

465 Коммитов

Автор SHA1 Сообщение Дата
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
Yuge Zhang 19cb631537
Mutable shortcut and enhancements (#5336) 2023-02-15 10:38:42 +08:00
Yuge Zhang 1160fa734c
Refactor nas.nn (Stage 2) - Choices (#5280) 2023-02-13 10:37:13 +08:00
liuzhe-lz 9e1a8e8ff7
Local Training Service V3 (#5243) 2023-02-08 07:20:28 +08:00
Yuge Zhang 48f2df5706
Refactor nas.nn (Stage 1) - MutableModule (#5279) 2023-02-07 09:41:52 +08:00
Yuge Zhang 74f13f31a4
Refactor graph-based model space (#5270) 2023-02-05 10:07:38 +08:00
J-shang 6f5dd231ae
[Compression] v2.5 stage 1: add setting & test (#5292) 2023-02-02 10:27:03 +08:00
Yuge Zhang f33e8e5f88
NAS executable model space (#5269) 2023-01-18 14:21:00 +08:00
Louis-J d9be0e6b14
fix(speedup): fix the error on jit translation of 'aten::to.dtype_layout' in pytorch <= 1.10 (#5315) 2023-01-16 21:15:12 +08:00
Louis-J 8418f8021e
feat(speedup): add the concrete_trace_utils, to extend the torch.fx feature to trace more models into fx.graph (#5121) 2023-01-16 13:01:51 +08:00
Yuge Zhang 13c0b3d6ee
NAS evaluator V3 (#5268) 2023-01-16 00:15:06 +08:00