TRT_LOGGER Verbosity: Severity.ERROR

 

TRT_LOGGER Verbosity: Severity.ERROR

 

https://github.com/NVIDIA/TensorRT/issues/289

 

# Start TensorRT 6 Container
nvidia-docker run -it -v ${PWD}:/mnt --workdir=/mnt nvcr.io/nvidia/tensorrt:19.12-py3

# Install dependencies
pip install torch==1.4 torchvision==0.5

# Download test model code
wget https://gist.githubusercontent.com/rmccorm4/b72abac18aed6be4c1725db18eba4930/raw/fbdb009152ef54abe7d8b23a9fd57a0f250e03e4/alexnet_onnx.py

# Create test model
python alexnet_onnx.py --opset=11

# Parse test model
trtexec --onnx=alexnet_fixed.onnx --explicitBatch

...
In node 0 (importModel): INVALID_GRAPH: Assertion failed: tensors.count(input_name)
&&&& FAILED TensorRT.trtexec # trtexec --onnx=alexnet_fixed.onnx --explicitBatch

你可能感兴趣的:(onnx)