search

NVIDIA / TensorRT

NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source components of TensorRT.
https://developer.nvidia.com/tensorrt
Apache License 2.0
10.87k stars 2.14k forks source link

QAT MaxCalibrator #4043

Closed AnnaTrainingG closed 3 months ago

AnnaTrainingG commented 4 months ago

Description

default calib is MaxCalibrator, so this code will never use in QAT? https://github.com/NVIDIA/TensorRT/blob/4575799a91f67c060cd34212a9a27d0264460071/tools/pytorch-quantization/examples/torchvision/classification_flow.py#L628

lix19937 commented 3 months ago

It depends on https://github.com/NVIDIA/TensorRT/blob/4575799a91f67c060cd34212a9a27d0264460071/tools/pytorch-quantization/examples/torchvision/classification_flow.py#L247