👨🏻💻 وقتی با GPU مدل رو آموزش میدی، یه مشکل بزرگ وجود داره:
✏️ زمانی که GPU داره کار میکنه، CPU بیکاره! وقتی هم CPU داره داده رو آماده میکنه، GPU بیکاره! این رفت و برگشتا کلی وقت تلف میکنه. اما راه حلش سادهتر از اون چیزیه که فکر میکنی!
💰 راه حل جادویی: با Memory Pinning کاری کن CPU و GPU مثل دو همکار حرفهای باهم همکاری کنن!
❓چطوری؟ فقط با ۲ قدم ساده!
🔢 قفل کردن داده تو رم (Pin Memory)
loader = DataLoader(..., pin_memory=True)
🔢 انتقال داده به GPU بدون معطلی! (Non-Blocking)
data = data.to(device, non_blocking=True)
🔴 چه اتفاقی میوفته؟
🔹 وقتی GPU داره روی بچِ اول کار میکنه، همزمان CPU بچِ دوم رو آماده میکنه! دیگه GPU منتظر نمیمونه. مدام کار میکنه و زمان آموزش تا ۵ برابر کم میشه!
🔻دوستانی که یادگیری عمیق کار کردن فقط متوجه میشن 😁
🌐 #یادگیری_عمیق #DeepLearning
➖➖➖➖➖➖➖➖➖➖➖➖➖