BokyLiu opened a new issue #4427: PReLU dimension mismatch error when 
converting ONNX model to TVM
URL: https://github.com/apache/incubator-tvm/issues/4427
 
 
   Hi, I try to convert a pytorch model to tvm via onnx intermediate model 
following this tutorial, but fail at prelu operation convertion which reports a 
dimension mismatch error:
   ```
   tensor type `Tensor[(64), float32]` has 1 dimensions, while `Tensor[(64, 1, 
1), float32]` has 3 dimensions; unable to unify: `Tensor[(64), float32]` and 
`Tensor[(64, 1, 1), float32]`;
   ```
   every single **prelu** in my model has same error. 
   BTW there has same warnings at begining:
   ```
   WARNING:root:Attribute momentum is ignored in relay.sym.batch_norm
   WARNING:root:Attribute momentum is ignored in relay.sym.batch_norm
   WARNING:root:Attribute momentum is ignored in relay.sym.batch_norm
   WARNING:root:Attribute momentum is ignored in relay.sym.batch_norm
   WARNING:root:Attribute momentum is ignored in relay.sym.batch_norm
   WARNING:root:Attribute momentum is ignored in relay.sym.batch_norm
   WARNING:root:Attribute momentum is ignored in relay.sym.batch_norm
   WARNING:root:Attribute momentum is ignored in relay.sym.batch_norm
   WARNING:root:Attribute momentum is ignored in relay.sym.batch_norm
   WARNING:root:Attribute momentum is ignored in relay.sym.batch_norm
   WARNING:root:Attribute momentum is ignored in relay.sym.batch_norm
   WARNING:root:Attribute momentum is ignored in relay.sym.batch_norm
   ```
   my tvm is 0.6. 
   Need help, tks.

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
[email protected]


With regards,
Apache Git Services

Reply via email to