هذه المدونة سوف تصف:
- ما هي المعلمات في PyTorch؟
- لماذا يحتاج المستخدمون إلى التحقق من معلمات النموذج؟
- كيفية عرض عدد معلمات النموذج في PyTorch؟
ما هي المعلمات في PyTorch؟
في PyTorch، ' nn.Module 'يتم استخدام فئة لتحديد النماذج. ويشمل جميع العمليات والطبقات التي يتكون منها النموذج. تحتوي كل طبقة على مجموعة من المعلمات. يتم تحديث المعلمات بشكل أساسي أثناء التدريب لتقليل الخطأ بين القيم الفعلية للنموذج والتنبؤات.
لماذا يحتاج المستخدمون إلى التحقق من معلمات النموذج؟
أثناء تدريب النموذج، يحتاج المستخدمون إلى معرفة عدد معلمات نموذجهم لأنه يتطلب الكثير من الذاكرة وقوة المعالجة. إذا كانوا على دراية بعدد معلمات النموذج، فيمكنهم بسهولة تقييم مقدار الذاكرة المطلوبة ومقدار الوقت الذي سيستغرقه التدريب مما يساعد المستخدمين على تحسين عملية التدريب الخاصة بهم وكذلك منع نفاد النظام فضاء.
كيفية عرض عدد معلمات النموذج في PyTorch؟
ال ' nn.Module 'الفصل لديه' حدود() 'الطريقة المستخدمة لعرض عدد معلمات النموذج في نموذج PyTorch. للحصول على جميع العناصر، ' رقم 1() 'يتم استخدام الطريقة.
لفهم المفهوم الذي تمت مناقشته مسبقًا، دعونا نلقي نظرة على الكود المقدم:
يستورد شعلة. ن.ن مثل ن.ن
فصل NNModel ( ن.ن. وحدة ) :
مواطنه __حار__ ( الذات ) :
ممتاز ( NNModel , الذات ) . __حار__ ( )
الذات . fc1 = ن.ن. خطي ( 10 , خمسون )
الذات . fc2 = ن.ن. خطي ( خمسون , 1 )
مواطنه إلى الأمام ( الذات , أنا ) :
أنا = الذات . fc1 ( أنا )
أنا = الذات . fc2 ( أنا )
يعود أنا
طرازى = NNModel ( )
t_params = مجموع ( ص. إعطاء اسم ( ) ل ص في طرازى. حدود ( ) )
مطبعة ( F 'إجمالي عدد المعلمات: {t_params}' )
في الكود المذكور أعلاه:
- أولاً، نحدد نموذجًا يحتوي على طبقتين خطيتين.
- ثم قم بإنشاء مثيل النموذج واستخدم ' حدود() 'طريقة لاسترداد جميع المعلمات.
- بعد ذلك، نطبق تعبير المولد لحساب جميع المعلمات من خلال جمع عدد عناصر كل معلمة.
- واخيرا اتصل على ' مطبعة() عبارة 'لعرض القيم الناتجة على الشاشة:
في الكود الموضح أعلاه، قمنا فقط بعرض العدد الإجمالي للمعلمات، إذا كنت ترغب في الحصول على اسم المعلمة وحجمها، فيمكن استخدام سطور التعليمات البرمجية التالية:
ل اسم , المعلمة في طرازى. State_dict ( ) . أغراض ( ) :مطبعة ( اسم , المعلمة. مقاس ( ) )
هنا:
- ' State_dict() ' هو كائن قاموس Python الذي يتم استخدامه لتخزين وتحميل النماذج من PyTorch.
- ' غرض() يتم استخدام الطريقة لإرجاع القائمة بجميع مفاتيح القاموس بالإضافة إلى القيم.
- ' مطبعة() 'يتم استخدام العبارة لطباعة اسم المعلمة وحجمها عن طريق تمرير' مقاس() 'الطريقة والمعلمة:
هذا كل شئ! لقد قمنا بتجميع أسهل طريقة لطباعة عدد معلمات النموذج في PyTorch.
خاتمة
في PyTorch، ' nn.Module يتم استخدام فئة 'لتعريف النماذج التي تتضمن جميع العمليات والطبقات التي يتكون منها النموذج. ال ' nn.Module 'الفصل لديه' حدود() 'الطريقة المستخدمة لعرض عدد معلمات النموذج في نموذج PyTorch. توضح هذه الكتابة طريقة طباعة عدد معلمات النموذج في PyTorch.