search

hunglc007 / tensorflow-yolov4-tflite

YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt and tflite
https://github.com/hunglc007/tensorflow-yolov4-tflite
MIT License
2.24k stars 1.24k forks source link

convert_tflite.py CUDA_ERROR_UNKNOWN: unknown error #307

Open p-agostinho opened 3 years ago

p-agostinho commented 3 years ago

Hello. First of all, thanks for the repo. I'm trying to convert a normal yolov4 weights to a tf model. I can run the first command, with 'save_model.py'. However, the simple 'python convert_tflite.py' command produces the following error:

F tensorflow/stream_executor/lib/statusor.cc:34] Attempting to fetch value instead of handling error Internal: failed to get device attribute 13 for device 0: CUDA_ERROR_UNKNOWN: unknown error

After the error, the execution is aborted and does not finish.

Do you know what might be the cause? Thanks.

p-agostinho commented 3 years ago

Update: I still get the error on my own computer, but managed to get it working on google colab with tensorflow 2.3.0. However, now I found an error related to int8 quantization of yolov4-tiny on the default dataset. Might create a new issue later