pytorch / TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
https://pytorch.org/TensorRT
BSD 3-Clause "New" or "Revised" License
2.41k stars 333 forks source link

fix: bug in vgg16_fp8_ptq example #2950

Closed zewenli98 closed 1 week ago

zewenli98 commented 1 week ago

Description

Fix a bug in vgg16_fp8_ptq.py example.

Fixes #2932

Type of change

Checklist: