I’m trying to implement a variant of Resnet in Jetson TX2.
I built and trained my model in Keras, froze the model to .pb, converted it to uff with convert-to-uff. All this happened successfully.
Then used the sampleUffMNIST.cpp as a reference to load the uff model for inference in Jetson. While loading it, I got the following error:
ERROR: add_3/add: all elementwise inputs must have same dimensions or follow the broadcasting rules ERROR: sample_uff: Unable to create engine ERROR: sample_uff: Model load failed
This error comes from the add_3 layer, that adds the outputs of two Conv+activation layers. From model summary in keras, I can see those two Conv layers have the same output shape: (?,128,23,40) and the add_3 layer also has the same output shape.
What is wrong here? Isn’t there any way to implement resnets in Jetson?
Any kind of help is appreciated…
Here’s my model architecture: https://drive.google.com/open?id=1Dxc3FJkUyNxiFj0xaYWKy1rwrfreGtV-