Hi,
I am using tensorrt on jetson nano.
I am trying to run the mnist pytorch example code(provided with tensorrt) in tensorrt folder but it doesn’t run.
changing TRT_LOGGER = trt.Logger(trt.Logger.WARNING) to TRT_LOGGER = trt.Logger(trt.Logger.INFO) I am getting the following output…
[TensorRT] INFO: Original: 7 layers
[TensorRT] INFO: After dead-layer removal: 7 layers
[TensorRT] INFO: After scale fusion: 7 layers
[TensorRT] INFO: Fusing (Unnamed Layer* 4) [Fully Connected] with (Unnamed Layer* 5) [Activation]
[TensorRT] INFO: Fusing (Unnamed Layer* 4) [Fully Connected] + (Unnamed Layer* 5) [Activation] with (Unnamed Layer* 6) [Fully Connected]
[TensorRT] INFO: After vertical fusions: 5 layers
[TensorRT] INFO: After swap: 5 layers
[TensorRT] INFO: After final dead-layer removal: 5 layers
[TensorRT] INFO: After tensor merging: 5 layers
[TensorRT] INFO: After concat removal: 5 layers
[TensorRT] INFO: Graph construction and optimization completed in 0.00094719 seconds.
Killed