Error Of ModelSpeedup - Githubissues

microsoft / nni

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

MIT License

13.97k stars 1.81k forks source link

Describe the bug: when using ModelSpeedup after quantization, an error was reported. I'm unsure whether the problem should be attributed to the models themselves. But what I used are all regular models and the DoReFA quantizer doesn't relate to the layer where the bug appears. Can you help me with this?

Environment:

NNI version: 2.10
Training service (local|remote|pai|aml|etc): remote
Python version: 3.8
PyTorch version: 1.13.1
Cpu or cuda version: 11.7

Reproduce the problem

Code|Example:

config_list = [{
    'quant_types': ['weight'],
    'quant_bits': {'weight': 8},
    'op_types': ['Conv2d']
}]

Bug:

microsoft / nni

Error Of ModelSpeedup #5306