Repository navigation
TensorRT compatible DocShadow ONNX Models
This release provides exported ONNX models of the DocShadow models (ONNX opset version 12).
docshadow_{dataset}.onnx: DocShadow trained ondataset.
Environment Compatibility & Reproducibility
These models were exported and tested with the following framework versions:
CUDA==11.8cuDNN==8.9.4TensorRT==8.6.1torch==2.0.1torchvision==0.15.2onnx==1.14.0onnxconverter-common==1.13.0onnxruntime-gpu==1.15.1