There is an if else statement in the model I built myself, and an error will appear when the model is accelerated during model pruning. How should I solve it?

microsoft / nni

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

https://nni.readthedocs.io

MIT License

13.88k stars 1.81k forks source link

There is an if else statement in the model I built myself, and an error will appear when the model is accelerated during model pruning. How should I solve it? #5757

Open xxiayy opened 3 months ago

xxiayy commented 3 months ago

Describe the issue:

Environment:

NNI version:
Training service (local|remote|pai|aml|etc):
Client OS:
Server OS (for remote mode only):
Python version:
PyTorch/TensorFlow version:
Is conda/virtualenv/venv used?:
Is running in Docker?:

Configuration:

Experiment config (remember to remove secrets!):
Search space:

Log message:

nnimanager.log:
dispatcher.log:
nnictl stdout and stderr:

How to reproduce it?: