Error to generate wts file

marcoslucianops / DeepStream-Yolo

NVIDIA DeepStream SDK 7.0 / 6.4 / 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 / 5.1 implementation for YOLO models

MIT License

1.42k stars 350 forks source link

Error to generate wts file #113

Closed juanfeyero closed 2 years ago

juanfeyero commented 2 years ago

I followed the step. I installed PyTorch 1.7.0. But, it showed me the following error when I run : python3 gen_wts_yolor.py -w yolor_csp.pt -c cfg/yolor_csp.cfg Traceback (most recent call last): File "gen_wts_yolor.py", line 5, in from utils.torch_utils import select_device ModuleNotFoundError: No module named 'utils'

Please help me.

marcoslucianops commented 2 years ago

you need to copy the gen_wts_yolor.py to the yolor folder