#5805·nni

About encountering exception while using ConvTranspose2d in nni pruning

Author: ChristopherCaoCreated Sep 5, 2024Updated Sep 5, 2024

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 image 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 image 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?: