error while executing in int8 mode

lewes6369 / TensorRT-Yolov3

TensorRT for Yolov3

MIT License

487 stars 165 forks source link

error while executing in int8 mode #13

Closed shiva13425 closed 5 years ago

shiva13425 commented 5 years ago

While executing: ./install/runYolov3 --caffemodel=./yolov3_416.caffemodel --prototxt=./yolov3_416.prototxt --input=./test.jpg --W=416 --H=416 --class=80 --mode=int8 --calib=./calib_sample.txt

ERROR: load batch 0 to 0 load batch 1 to 1 runYolov3: NvPluginYOLO.cu:58: virtual void nvinfer1::plugin::PReLU::configure(const nvinfer1::Dims, int, const nvinfer1::Dims, int, int): Assertion `mBatchDim == 1' failed.

Kindly share your comments. Thanks

lewes6369 commented 5 years ago

Just run one more time, using the cache it will be OK. I thought that is a bug in PReLU by tensorRT implement. Testing other models is OK once.

shiva13425 commented 5 years ago

@lewes6369 Thanks . It worked.

haithink commented 5 years ago

yes, run the program with the same arguments and there will be no error.