search

Tianxiaomo / pytorch-YOLOv4

PyTorch ,ONNX and TensorRT implementation of YOLOv4
Apache License 2.0
4.47k stars 1.49k forks source link

calibration to int8 INTERNAL ERROR: Assertion failed: d.nbDims >= 1 #530

Open daliel opened 2 years ago

daliel commented 2 years ago

Tensorrt 7.0.0.11-1+cuda10.2 CUDA 10.2 Ubuntu 18.04 python3.6 torch 1.4.0 onnxruntime 1.1.1

code for calibration from: https://github.com/Tianxiaomo/pytorch-YOLOv4/issues/503

[TensorRT] ERROR: FAILED_EXECUTION: std::exception [TensorRT] WARNING: Explicit batch network detected and batch size specified, use execute without batch size instead. [TensorRT] INTERNAL ERROR: Assertion failed: d.nbDims >= 1 ../rtSafe/safeHelpers.cpp:419 Aborting...

Lenan22 commented 1 year ago

Tensorrt 7.0.0.11-1+cuda10.2 CUDA 10.2 Ubuntu 18.04 python3.6 torch 1.4.0 onnxruntime 1.1.1

code for calibration from: #503

[TensorRT] ERROR: FAILED_EXECUTION: std::exception [TensorRT] WARNING: Explicit batch network detected and batch size specified, use execute without batch size instead. [TensorRT] INTERNAL ERROR: Assertion failed: d.nbDims >= 1 ../rtSafe/safeHelpers.cpp:419 Aborting... https://github.com/Tianxiaomo/pytorch-YOLOv4/issues/530#issue-1240142840