Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

推理的时候报错是为什么啊? #24

Open
shuishen49 opened this issue Mar 28, 2024 · 0 comments
Open

推理的时候报错是为什么啊? #24

shuishen49 opened this issue Mar 28, 2024 · 0 comments

Comments

@shuishen49
Copy link

YOLOv5 🚀 2024-3-27 Python-3.9.16 paddle-2.4.1 CUDA:0 (NVIDIA A100-SXM4-40GB, 40337MiB)

W0328 18:14:35.827714 55102 gpu_resources.cc:61] Please NOTE: device: 0, GPU Compute Capability: 8.0, Driver API Version: 12.0, Runtime API Version: 11.2
W0328 18:14:35.830621 55102 gpu_resources.cc:91] device: 0, cuDNN Version: 8.2.
Fusing layers...
Model summary: 211 layers, 20893344 parameters, 64656 gradients, 48.1 GFLOPs
Traceback (most recent call last):
File "/home/aistudio/yolov5-Paddle/detect.py", line 255, in
main(opt)
File "/home/aistudio/yolov5-Paddle/detect.py", line 250, in main
run(**vars(opt))
File "", line 2, in run
File "/opt/conda/envs/python35-paddle120-env/lib/python3.9/site-packages/paddle/fluid/dygraph/base.py", line 375, in _decorate_function
return func(*args, **kwargs)
File "/home/aistudio/yolov5-Paddle/detect.py", line 149, in run
det[:, :4] = scale_boxes(im.shape[2:], det[:, :4], im0.shape).round()
File "/opt/conda/envs/python35-paddle120-env/lib/python3.9/site-packages/paddle/fluid/dygraph/varbase_patch_methods.py", line 753, in getitem
return self._getitem_index_not_tensor(item)
ValueError: (InvalidArgument) Too many indices (2) for tensor of dimension 1.
[Hint: Expected valid_indexs <= rank == true, but received valid_indexs <= rank:0 != true:1.] (at /paddle/paddle/fluid/pybind/slice_utils.h:306)
我的亲娘哎。训练了2天没有结果。。。。。。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant