Skip to content

[6090146] Added support inp->DQ graphs#34

Open
gcunhase wants to merge 2 commits into
NVIDIA:mainfrom
gcunhase:dev/gcunhasergio/support_dq_input
Open

[6090146] Added support inp->DQ graphs#34
gcunhase wants to merge 2 commits into
NVIDIA:mainfrom
gcunhase:dev/gcunhasergio/support_dq_input

Conversation

@gcunhase
Copy link
Copy Markdown
Collaborator

Overview

Type: Feature enhancement

Changes:

Original QDQ-Translated (Before) QDQ-Translated (After)
resnet_original resnet_qdq_translated_before resnet_qdq_translated_after

This also

Steps to repro

  1. Install ModelOpt:
$ pip install nvidia-modelopt[onnx]
  1. Download example model: resnet50-v1-12.onnx

  2. Obtained explicitly-quantized (EQ) model:

$ python -m modelopt.onnx --onnx_path=resnet50-v1-12.onnx --direct_io_types
  1. Obtain DLA-compatible model and cache / args files with QDQ-Translator:
$ python tools/qdq-translator/qdq_translator.py -i resnet50-v1-12.quant.onnx -o translated

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant