使用yolov5xTPH，显存占用过大

Gumpest / YOLOv5-Multibackbone-Compression

YOLOv5 Series Multi-backbone(TPH-YOLOv5, Ghostnet, ShuffleNetv2, Mobilenetv3Small, EfficientNetLite, PP-LCNet, SwinTransformer YOLO), Module(CBAM, DCN), Pruning (EagleEye, Network Slimming), Quantization (MQBench) and Deployment (TensorRT, ncnn) Compression Tool Box.

995 stars 202 forks source link

使用yolov5xTPH，显存占用过大 #11

Closed tonyskypc closed 2 years ago

tonyskypc commented 2 years ago

runtimeerror:cuda out of memory.tried to allocate 198.75GiB

这个要用这样大的显存才能跑起来，我有4张v100 32G都不行，请问是什么原因。

Gumpest commented 2 years ago

您好，请用最新的yaml文件尝试一下；之前yolov5x无法兼容TPH标准结构。

tonyskypc commented 2 years ago

就是刚刚用的您的最新代码，yaml看了一下是您4天前提交的那份。

Gumpest commented 2 years ago

您好，我今天更新了一下；请git pull一下

tonyskypc commented 2 years ago

更新今天最新版本后，已经可以训练起来了，非常感谢作者。