nni
nni copied to clipboard
An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.
**Describe the issue**: When I tried to prune some detection models, I encountered an error that did not support `aten:: randn`. And this operator is usually used as the generation...
**Describe the issue**: Hi, when using model speed up, the model size and model's channel output did reduce, however, the inference did not improve. The code I use to test...
**Describe the issue**: When I was using nni to tune my model, this error occur after the experiment started for a period of time: `Dispatcher stream error, tuner may have...
nni-manager log file: ``` [2022-07-20 07:49:27] ERROR (tuner_command_channel.WebSocketChannel) Error: Error: tuner_command_channel: Tuner closed connection at WebSocket. (/home/nzy/miniconda3/envs/pyg1/lib/python3.8/site-packages/nni_node/core/tuner_command_channel/websocket_channel.js:41:49) at WebSocket.emit (node:events:538:35) at WebSocket.emitClose (/home/nzy/miniconda3/envs/pyg1/lib/python3.8/site-packages/nni_node/node_modules/express-ws/node_modules/ws/lib/websocket.js:246:10) at Receiver.receiverOnFinish (/home/nzy/miniconda3/envs/pyg1/lib/python3.8/site-packages/nni_node/node_modules/express-ws/node_modules/ws/lib/websocket.js:1037:20) at Receiver.emit (node:events:526:28) at...
**Spark Mlib support in NNI**
**What I would you like to be added**: It would be great if NNI had an *extension/framework for model selection*. **Why is this needed**: Model selection is a very relevant...
**What would you like to be added**: I work in an AI company, and we hope that NNI can not only save JSON representing the optimal model, but also the...
**Describe the issue**: I want to run DARTS examples in multiple GPUs, so I wrapped the model with DDP and shared data with Distributedsampler. However, I found the final 2...
**Describe the issue**: When I tried to use NAS, it reported 'ImportError: Cannot use a path to identify something from __main__.' and 'ValueError: Pickle too large when trying to dump...
**Describe the issue**: **Environment**: - NNI version: 2.2 - Training service (local|remote|pai|aml|etc): kubeflow - Client OS: ubuntu 18.04 - Server OS (for remote mode only): ubuntu 18.04 - Python version:...