tinyvision / DAMO-YOLO

DAMO-YOLO: a fast and accurate object detection method with some new techs, including NAS backbones, efficient RepGFPN, ZeroHead, AlignedOTA, and distillation enhancement.
Apache License 2.0
3.75k stars 470 forks source link

error for single GPU training #82

Closed zqyJason closed 1 year ago

zqyJason commented 1 year ago

Before Asking

Search before asking

Question

When I train damoyolo_tinynasL35_M on my custom dataset on single GPU(Tesla T4), there always throw this error:

RuntimeError: CUDA error: out of memory
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.

but actually my GPU has no processes occupied.

Additional

No response

cwhgn commented 1 year ago

This error may be caused by insufficient cuda memory. You can reduce the batch size to see if it works

zqyJason commented 1 year ago

This error may be caused by insufficient cuda memory. You can reduce the batch size to see if it works

I have tried it, but it doesn't work. It's really weird.