-
### Search before asking
- [X] I have searched the YOLOv5 [issues](https://github.com/ultralytics/yolov5/issues) and found no similar bug report.
### YOLOv5 Component
Training
### Bug
I have use…
-
Hi, I would like to train custom dataset with 1 label. I have followed the wiki and configured all the requirements. However, I got this error for my CUDA:
![Cuda Error](https://user-images.githubuse…
-
Contents removed for not adhering to the Ultralytics [Code of Conduct](https://docs.ultralytics.com/help/code_of_conduct/).
-
Hello Glenn et al.
I have a dataset with 512 x 512 image size and decent size boxes:
![ground_truth_droplets](https://user-images.githubusercontent.com/91646805/136671108-bcf65165-6af3-4d5f-926…
-
pytorch1.1 cuda9.0, run train get error:
```
Model Summary: 222 layers, 6.15291e+07 parameters, 6.15291e+07 gradients
Epoch Batch xy wh conf cls total nTarg…
-
## 🐛 Bug
Hi, When I follow the advise to save a onnx model. But onnx.checker.check_model(model) faild.
## To Reproduce
Steps to reproduce the behavior:
1. Set ONNX_EXPORT to True
2. have torc…
-
When I trained my datas using multi-GPU , freeze_backbone=True, It reported wrongs:
Traceback (most recent call last):
File "train.py", line 257, in
num_workers=opt.num_workers
File "trai…
-
## ❔Question
Hi. I just pull the git repo today and I got the following error while running test.py. Looks like the "model.load_state_dict()" has some issue. Anyone knows why this happens? Thanks!
…
-
The bug
I am trying to export my model to ONNX. However, I have the error:
TypeError: 'YOLOLayer' object does not support indexing
Steps to reproduce the behavior:
1. Set: ONNX_EXPORT = True
…
-
I have two 1080ti graphics cards. When I train with one (just one card installed in the computer) everything goes fine. But when I put another card into the computer, the training process freezes at t…