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

2961 Коммитов

Автор SHA1 Сообщение Дата
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
HuangLuGuang eaa006846a
add Python API to kill a trail job (#5411) 2023-03-02 11:02:59 +08:00
QuanluZhang 20c5238937
[quant] add comments (#5410) 2023-03-01 14:59:50 +08:00
Yuge Zhang d5c14d8621
NAS profiler (stage 1) - interface (#5361) 2023-03-01 14:20:37 +08:00
Yuge Zhang c98e1f38f6
NAS strategy (stage 1) - interface (#5371) 2023-03-01 10:48:12 +08:00
Yuge Zhang e94eb99ce2
Rewrite resume and view for experiment (#5365) 2023-03-01 10:44:15 +08:00
Yuge Zhang 1582ab6d31
[MAGIC] Debug MacOS simple integration test (#5413) 2023-03-01 09:52:58 +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
Lijiaoa bfacbc28de
Standardize and improve styles (#5402) 2023-02-27 10:39:32 +08:00
Yuge Zhang 0dce45a845
Tuner Command Channel (#5364) 2023-02-27 10:35:37 +08:00
Yuge Zhang 4f92f30a0f
NAS execution engine (stage 1) - Engine interface (#5358) 2023-02-24 10:51:26 +08:00
Yuge Zhang 03944bd443
Refactor nas.nn (Stage 4) - Model Space Hub (#5282) 2023-02-24 10:51:05 +08:00
Yuge Zhang dd4e5909c3
New trial runner and import trial command channel (#5398) 2023-02-23 14:14:33 +08:00
liuzhe-lz 99f9c71b51
[Remote v3] Stage 0 - WebSocket command channel server (#5346) 2023-02-23 11:53:16 +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
Lijiaoa c4564ab5ba
upgrade axios,node-sass version (#5352) 2023-02-17 10:40:46 +08:00
liuzhe-lz b8f6a894a7
Fix bugs in training service v3 shim layer (#5357) 2023-02-17 10:39:29 +08:00
Yuge Zhang 1aa5c251ac
Refactor nas.nn (Stage 3) - Cell and Repeat (#5281) 2023-02-17 10:39:05 +08:00
J-shang badf49dfee
[Compression] v2.5 stage 5: add QAT quantizer (#5296) 2023-02-16 18:54:14 +08:00
Yuge Zhang 19cb631537
Mutable shortcut and enhancements (#5336) 2023-02-15 10:38:42 +08:00
Lijiaoa 9b59d6d797
Select all & select none in customized model and some bug bash work (#5297) 2023-02-13 10:39:51 +08:00
Yuge Zhang 1160fa734c
Refactor nas.nn (Stage 2) - Choices (#5280) 2023-02-13 10:37:13 +08:00
Super Daniel efe1246323
Fix issue #5299. (#5343) 2023-02-10 10:49:44 +08:00
Yuge Zhang 8371d228f5
Fix logo for documentation (#5337) 2023-02-08 10:42:02 +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
J-shang 15ce74957d
[Compression] v2.5 stage 4: add compressor (#5295) 2023-02-06 17:20:51 +08:00
Yuge Zhang db08e5617e
Fix jquery dependency in docs and add sphinx-comment (#5326) 2023-02-06 10:44:29 +08:00
Yuge Zhang 74f13f31a4
Refactor graph-based model space (#5270) 2023-02-05 10:07:38 +08:00
J-shang a722af68f7
[Compression] v2.5 stage 3: add wrapper (#5294) 2023-02-03 10:41:35 +08:00
J-shang e142d6cd78
[Compression] v2.5 stage 2: add target space and config (#5293) 2023-02-03 09:33:16 +08:00
J-shang 6f5dd231ae
[Compression] v2.5 stage 1: add setting & test (#5292) 2023-02-02 10:27:03 +08:00
Bonytu 660f7c1bf5
ddp_example_with_taylorFo (#5289) 2023-01-30 11:02:55 +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
Yuge Zhang 8bcd9ed23f
Rerun flaky tests (#5316) 2023-01-13 10:40:34 +08:00
J-shang 4d93790d66
[Compression] v2.5 stage 0: copy useful utils from old version (#5291) 2023-01-10 10:04:55 +08:00
Yuge Zhang e43272a972
[Serializer] `_load`, `_dump` and `_unwrap_parameter` (#5267) 2023-01-04 13:43:23 +08:00