Skip to content
This repository has been archived by the owner on Sep 18, 2024. It is now read-only.

About encountering exception while using ConvTranspose2d in nni pruning #5805

Open
ChristopherCao opened this issue Sep 5, 2024 · 0 comments

Comments

@ChristopherCao
Copy link

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

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant