Skip to content

v1.0.0: DocShadow-ONNX-TensorRT

Latest

Choose a tag to compare

@fabio-sim fabio-sim released this 31 Aug 17:34
· 2 commits to main since this release
e92f020

TensorRT compatible DocShadow ONNX Models

This release provides exported ONNX models of the DocShadow models (ONNX opset version 12).

  • docshadow_{dataset}.onnx: DocShadow trained on dataset.

Environment Compatibility & Reproducibility

These models were exported and tested with the following framework versions:

  • CUDA==11.8
  • cuDNN==8.9.4
  • TensorRT==8.6.1
  • torch==2.0.1
  • torchvision==0.15.2
  • onnx==1.14.0
  • onnxconverter-common==1.13.0
  • onnxruntime-gpu==1.15.1