An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.
Training service (local|remote|pai|aml|etc): local
Python version: 3.10
PyTorch version: 1.13
Cpu or cuda version: Cuda/Cpu
Reproduce the problem
How to reproduce: It seems like the classTransformerEncoderLayer is not currently supported by NNI. I am attempting to prune the network RepNet . Can someone help adding the implementation for that?
Describe the bug:
Environment:
Reproduce the problem
TransformerEncoderLayeris not currently supported by NNI. I am attempting to prune the network RepNet . Can someone help adding the implementation for that?