1. update pytorch-quantization to 2.1.3
2. update SD to support torch 2.x
3. update docker container
4. misc fixed in samples
Signed-off-by: Vincent Huang <vincenth@nvidia.com>
Also adds goldenIO testcases: one for class agnostic NMS and another for class aware NMS.
With this, golden_io_snippet.py is also modified to support checking for multiple
plugin outputs.
Adds `class_agnostic` parameter to `EfficientNMS_TRT` plugin:
- Some object detection networks/architectures like YOLO series need to
use class-independent NMS operations.
- If `class_agnostic` is enabled, class-independent NMS is performed;
otherwise, different classes would do NMS separately.
Signed-off-by: Samurdhi Karunaratne <skarunaratne@nvidia.com>
Signed-off-by: Shuyue Lan <shuyuel@nvidia.com>
Signed-off-by: Rajeev Rao <rajeevrao@nvidia.com>
Update tools including onnx-graphsurgeon(v0.3.25), Polygraphy(v0.43.1), tensorflow-quantization(v0.2.0)
Update onnx-tensorrt(8.5GA)
Mass merge sample, plugin, demo, python changes for TRT-8.5 GA release
Update demoBERT benchmark data for TensorRT 8.5.1
Signed-off-by: Simeng Liu <simengl@nvidia.com>
Signed-off-by: Simeng Liu <simengl@nvidia.com>
Vincent Huang