-
## ❔Question
Hello, i am trying use 4 yolov5m models. I have 4gb GPU Memory but while i was deploying models with pytorch hub it just uses 1.2 gb cuda memory is it possible the increase this allocate…
-
## ❔Question
python train.py --batch-size 64 --data ./data/car.yaml --cfg ./models/car_m.yaml --weights 'weights/yolov5m.pt' --device 0,1 --epoch 50
Using CUDA device0 _CudaDeviceProperties(name=…
-
## 🐛 Bug
I get an `Expected all tensors to be on the same device` error when using `$ python export.py --weight yolov5m.pt --include onnx --device 0`
## To Reproduce (REQUIRED)
Input:
```
…
-
I am running some dummy data through the model as follows:
```
import torch
image = torch.zeros(1, 3, 640, 640).to('cuda:0')
model = torch.load('yolov5.pt', map_location=map_location)['model']…
dagap updated
2 weeks ago
-
### Search before asking
- [X] I have searched the YOLOv5 [issues](https://github.com/ultralytics/yolov5/issues) and [discussions](https://github.com/ultralytics/yolov5/discussions) and found no simi…
-
## ❔Question
Hi All,
I have custom trained model for both yolov5 release 4 and release 6 using yolov5m.pt weights given to them. When I check the inference using Nvidia T4 GPU I get inference for …
-
### Search before asking
- [X] I have searched the YOLOv5 [issues](https://github.com/ultralytics/yolov5/issues) and [discussions](https://github.com/ultralytics/yolov5/discussions) and found no simi…
-
作者大大你好，谢谢你的开源，请问这个代码对yolov5.5的m模型适用吗？如果对网络结构进行了修改，请问又应该再如何进行剪枝呢？请问可以提供一个思路吗
-
## ❔Question
I have been training yolov5m model on 300 epochs, but during 107 epoch, my internet went off, so how to restore the trained model so that it can start from 107 epoch? Is there any way?
…
-
### Search before asking
- [X] I have searched the YOLOv5 [issues](https://github.com/ultralytics/yolov5/issues) and found no similar bug report.
### YOLOv5 Component
Export
### Bug
I'm trying t…