nvidia@tegra-ubuntu:~/TensoRT_C++/ForAGX/Aarch64/bin$ ./sample_onnx_mnist
******* MY initializeSampleParams: params.onnxFileName = yolov4_1_3_512_512_static.onnx
Building and running a GPU inference engine for Onnx MNIST
**************** build() ************************
 ******** constructNetwork() parseFromFile() ******************
----------------------------------------------------------------
Input filename:   ../data/mnist/yolov4_1_3_512_512_static.onnx
ONNX IR version:  0.0.6
Opset version:    11
Producer name:    pytorch
Producer version: 1.7
Domain:           
Model version:    0
Doc string:       
----------------------------------------------------------------
WARNING: ONNX model has a newer ir_version (0.0.6) than this parser was built against (0.0.3).
[06/09/2021-15:35:58] [E] [TRT] (Unnamed Layer* 0) [Convolution]: at least 4 dimensions are required for input
[06/09/2021-15:35:58] [I] [TRT] 2400:Conv -> 
[06/09/2021-15:35:58] [I] [TRT] 651:Softplus -> 
[06/09/2021-15:35:58] [I] [TRT] 652:Tanh -> 
[06/09/2021-15:35:58] [I] [TRT] 653:Mul -> 
While parsing node number 4 [Conv -> "2403"]:
3
--- Begin node ---
input: "653"
input: "2404"
input: "2405"
output: "2403"
name: "Conv_4"
op_type: "Conv"
attribute {
  name: "dilations"
  ints: 1
  ints: 1
  type: INTS
}
attribute {
  name: "group"
  i: 1
  type: INT
}
attribute {
  name: "kernel_shape"
  ints: 3
  ints: 3
  type: INTS
}
attribute {
  name: "pads"
  ints: 1
  ints: 1
  ints: 1
  ints: 1
  type: INTS
}
attribute {
  name: "strides"
  ints: 2
  ints: 2
  type: INTS
}

--- End node ---
ERROR: /home/nvidia/TensoRT_C++/REMOVE/TensorRT/parsers/onnx/builtin_op_importers.cpp:757 In function importConv:
[8] Assertion failed: (nbSpatialDims == 2 && kernel_weights.shape.nbDims == 4) || (nbSpatialDims == 3 && kernel_weights.shape.nbDims == 5)
nvidia@tegra-ubuntu:~/TensoRT_C++/ForAGX/Aarch64/bin$