search

airockchip / rknn_model_zoo

Apache License 2.0
1.05k stars 192 forks source link

量化时候一直卡着 #4

Closed nobody-cheng closed 3 months ago

nobody-cheng commented 2 years ago

centos7 rtx2080ti， cuda 11.1 cudnn 8 使用如下命令onnx转rknn模型，开启量化，量化图片3张

CUDA_VISIBLE_DEVICES=7 python ../../../../../common/rknn_converter/rknn_convert.py --yml_path ./yolov5s_convert.yml --compute_convert_loss --eval_perf

然后一直这里卡着不动 image