Skip to content
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
@ChristopherCao

Description

@ChristopherCao

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions