הבלוג הזה יתאר:
- מהם הפרמטרים ב- PyTorch?
- מדוע משתמשים צריכים לבדוק את הפרמטרים של המודל?
- כיצד להציג את מספר פרמטרי המודל ב- PyTorch?
מהם הפרמטרים ב- PyTorch?
ב- PyTorch, ה' nn.מודול מחלקה משמשת להגדרת המודלים. הוא כולל את כל הפעולות והשכבות המרכיבות את המודל. כל שכבה מכילה קבוצה של פרמטרים. הפרמטרים מתעדכנים בעצם תוך כדי אימון כדי למזער את השגיאה בין הערכים והתחזיות בפועל של המודל.
מדוע משתמשים צריכים לבדוק את הפרמטרים של המודל?
בזמן אימון המודל, המשתמשים צריכים לדעת על מספר הפרמטרים של המודל שלהם מכיוון שהוא דורש הרבה זיכרון וכוח עיבוד. אם הם מכירים את מספר הפרמטרים של הדגם, הם יכולים להעריך בקלות את כמות הזיכרון שיידרש וכמה זמן ייקח לאימון, מה שעוזר למשתמשים לייעל את תהליך האימון שלהם, כמו גם למנוע מהמערכת להיגמר מֶרחָב.
כיצד להציג את מספר פרמטרי המודל ב- PyTorch?
ה ' nn.מודול ' בכיתה יש את ' פרמטרים() שיטה המשמשת לצפייה במספר פרמטרי המודל במודל PyTorch. כדי לקבל את כל האלמנטים, ' num1() נעשה שימוש בשיטה.
כדי להבין את הרעיון שנדון קודם לכן, בואו נסתכל על הקוד שסופק:
יְבוּא לפיד. nn כפי ש nn
מעמד NNModel ( nn. מודול ) :
def __חַם__ ( עצמי ) :
סוּפֶּר ( NNModel , עצמי ) . __חַם__ ( )
עצמי . fc1 = nn. ליניארי ( 10 , חמישים )
עצמי . fc2 = nn. ליניארי ( חמישים , 1 )
def קָדִימָה ( עצמי , אני ) :
אני = עצמי . fc1 ( אני )
אני = עצמי . fc2 ( אני )
לַחֲזוֹר אני
המודל שלי = NNModel ( )
t_params = סְכוּם ( ע. תן שם ( ) ל ע ב המודל שלי. פרמטרים ( ) )
הדפס ( ו 'מספר כולל של פרמטרים: {t_params}' )
בקוד האמור לעיל:
- ראשית, נגדיר מודל בעל שתי שכבות ליניאריות.
- לאחר מכן, צור את המופע של המודל והשתמש ב-' פרמטרים() שיטה לאחזר את כל הפרמטרים.
- לאחר מכן, אנו מיישמים את ביטוי המחולל כדי לחשב את כל הפרמטרים על ידי סיכום מספר האלמנטים של כל פרמטר.
- לבסוף, התקשר ל' הדפס() ' הצהרה כדי להציג את הערכים המתקבלים על המסך:
בקוד המתואר לעיל, הצגנו רק את המספר הכולל של הפרמטרים, אם ברצונך לקבל את השם והגודל של הפרמטר, ניתן להשתמש בשורות הקוד הבאות:
ל שֵׁם , param ב המודל שלי. גזר דין_מדינה ( ) . פריטים ( ) :הדפס ( שֵׁם , param. גודל ( ) )
כאן:
- ' state_dict() ” הוא אובייקט מילון Python המשמש לאחסון וטעינת מודלים מ- PyTorch.
- ' פריט() השיטה משמשת להחזרת הרשימה עם כל מפתחות המילון יחד עם ערכים.
- ' הדפס() הצהרת ' משמשת להדפסת השם והגודל של הפרמטר על ידי העברת ה-' גודל() ' שיטה ופרמטר:
זה הכל! ריכזנו את הדרך הקלה ביותר להדפיס את מספר פרמטרי הדגם ב- PyTorch.
סיכום
ב- PyTorch, ה' nn.מודול ” class משמשת להגדרת המודלים הכוללים את כל הפעולות והשכבות המרכיבות את המודל. ה ' nn.מודול ' בכיתה יש את ' פרמטרים() שיטה המשמשת לצפייה במספר פרמטרי המודל במודל PyTorch. כתיבה זו הדגים את השיטה להדפסת מספר פרמטרי הדגם ב- PyTorch.