خطای CUDA Out of Memory هنگام آموزش مدل YOLOv8

حسام دهقان

۱ ماه پیش

gradient accumulation را فعال کنید تا با بچ‌سایز کوچک‌تر همان اثر بچ بزرگ را بگیرید.

مهسا وحیدی

۱ ماه پیش

mixed precision (AMP) را هم روشن کنید؛ مصرف حافظه را به‌طور محسوسی کم می‌کند.

رضا توکلی

۱ ماه پیش

اگر باز هم خطا داد، اندازهٔ ورودی (imgsz) را پایین بیاورید و مطمئن شوید چیزی روی GPU باقی نمانده باشد (torch.cuda.empty_cache).