This repository was archived by the owner on Sep 18, 2024. It is now read-only.
This repository was archived by the owner on Sep 18, 2024. It is now read-only.
About encountering exception while using ConvTranspose2d in nni pruning #5805
Open
Description
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:
- NNI version: 3.0
- Training service (local|remote|pai|aml|etc):
- Client OS:
- Server OS (for remote mode only): Ubuntu
- Python version: 3.9.10
- PyTorch/TensorFlow version: PyTorch version: 1.7.1
- Is conda/virtualenv/venv used?:yes
- Is running in Docker?:no
Configuration:
- Experiment config (remember to remove secrets!):
- Search space:
Log message:
- nnimanager.log:
- dispatcher.log:
- nnictl stdout and stderr:
How to reproduce it?:
Metadata
Metadata
Assignees
Labels
No labels