An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.
When using ConvTranspose2d in config_list to prune my models, I encountered an exception: Unsupported types for ConvTranspose2d.
However, when I check the library files, I see this
in nni/compression/speedup/mask_conflict.py, it deals with ConvTranspose2d here,
whereas does not deal with ConvTranspose2d here, which leads to exception for anyone who uses ConvTranspose2d in their codes
However, I have no idea to solve this by coding myself, so I report a bug here.
Hope to see the solution ASAP, thanks!
Describe the issue: Report an bug.
When using ConvTranspose2d in config_list to prune my models, I encountered an exception: Unsupported types for ConvTranspose2d. However, when I check the library files, I see this
in nni/compression/speedup/mask_conflict.py, it deals with ConvTranspose2d here,
whereas does not deal with ConvTranspose2d here, which leads to exception for anyone who uses ConvTranspose2d in their codes
However, I have no idea to solve this by coding myself, so I report a bug here.
Hope to see the solution ASAP, thanks!
Environment:
Configuration:
Log message:
How to reproduce it?: