ERROR: UFFParser: Parser error: gender_resnet0_batchnorm0/mul: Invalid scale mode, nbWeights: 3

AI & Data Science Deep Learning (Training & Inference) TensorRT
1

Hi all,

I’m facing a problem of inference a UFF model by using TensorRT on my TX2. The error message is like following.

ERROR: UFFParser: Parser error: gender_resnet0_batchnorm0/mul: Invalid scale mode, nbWeights: 3
ERROR: sample_uff_mnist: Fail to parse
ERROR: sample_uff_mnist: Model load failed

Anyone know how to solve this problem? I have seen a topic with same problem but it didn’t provide the solution. If needed, I can provide my model.

My environment:
TX2
TensorRT 4.1.3
CUDA9.0

2

I have met a similar error, and I think you can check the multiply operation responding to “gender_resnet0_batchnorm0/mul”.

It is supported in tensorflow if two multipliers with shape [1,32, 3, 2] and [3,2], but it is not supported in tensorrt. Then you can change the shape of the second multiplier to [1, 32, 3, 2], now you have two multiplier with shape[1, 32, 3, 2] and [1, 32, 3, 2], then the multiply operation can be supported in tensorrt, and the result will be the same.