nni icon indicating copy to clipboard operation
nni copied to clipboard

Published 20 hours ago • verified publisher icon microsoft

Unsupported Layer nn.TransformerEncoderLayer

Open saeedashrraf opened this issue 1 year ago • 0 comments

Describe the bug:

Environment:

  • NNI version: 3.0
  • Training service (local|remote|pai|aml|etc): local
  • Python version: 3.10
  • PyTorch version: 1.13
  • Cpu or cuda version: Cuda/Cpu

Reproduce the problem

  • How to reproduce: It seems like the classTransformerEncoderLayer is not currently supported by NNI. I am attempting to prune the network RepNet . Can someone help adding the implementation for that?

saeedashrraf avatar Dec 12 '23 21:12 saeedashrraf