Tento blog bude popisovať:
- Aké sú parametre v PyTorch?
- Prečo musia používatelia kontrolovať parametre modelu?
- Ako zobraziť počet parametrov modelu v PyTorch?
Aké sú parametre v PyTorch?
V PyTorch je „ nn.Modul ” trieda sa používa na definovanie modelov. Zahŕňa všetky operácie a vrstvy, ktoré tvoria model. Každá vrstva obsahuje súbor parametrov. Parametre sa v podstate aktualizujú počas tréningu, aby sa minimalizovala chyba medzi skutočnými hodnotami modelu a predpoveďami.
Prečo musia používatelia kontrolovať parametre modelu?
Počas trénovania modelu musia používatelia vedieť o počte parametrov svojho modelu, pretože to vyžaduje veľa pamäte a výpočtového výkonu. Ak sú oboznámení s množstvom parametrov modelu, môžu ľahko vyhodnotiť množstvo pamäte, ktoré bude potrebné a koľko času zaberie tréning, čo pomáha používateľom optimalizovať ich tréningový proces, ako aj zabrániť vyčerpaniu systému. priestor.
Ako zobraziť počet parametrov modelu v PyTorch?
' nn.Modul „trieda má „ parametre() ” metóda, ktorá sa používa na zobrazenie počtu parametrov modelu v modeli PyTorch. Ak chcete získať všetky prvky, „ num1() “ používa sa metóda.
Aby sme pochopili predtým diskutovaný koncept, pozrime sa na poskytnutý kód:
importovať fakľa. nn ako nn
trieda NNModel ( nn. modul ) :
def __horúce__ ( seba ) :
Super ( NNModel , seba ) . __horúce__ ( )
seba . fc1 = nn. Lineárne ( 10 , päťdesiat )
seba . fc2 = nn. Lineárne ( päťdesiat , 1 )
def dopredu ( seba , i ) :
i = seba . fc1 ( i )
i = seba . fc2 ( i )
vrátiť i
môj_model = NNModel ( )
t_params = súčet ( p. dať meno ( ) pre p v môj_model. parametre ( ) )
vytlačiť ( f 'Celkový počet parametrov: {t_params}' )
Vo vyššie uvedenom kóde:
- Najprv definujeme model, ktorý má dve lineárne vrstvy.
- Potom vygenerujte inštanciu modelu a použite „ parametre() ” metóda na získanie všetkých parametrov.
- Ďalej použijeme výraz generátora na výpočet všetkých parametrov sčítaním počtu prvkov každého parametra.
- Nakoniec zavolajte na „ tlačiť () ” na zobrazenie výsledných hodnôt na obrazovke:
Vo vyššie popísanom kóde sme zobrazili iba celkový počet parametrov, ak chcete získať názov a veľkosť parametra, môžete použiť nasledujúce riadky kódu:
pre názov , param v môj_model. state_dict ( ) . položky ( ) :vytlačiť ( názov , param. veľkosť ( ) )
Tu:
- “ state_dict() ” je objekt slovníka Pythonu, ktorý sa používa na ukladanie a načítanie modelov z PyTorch.
- “ položka() ” metóda sa používa na vrátenie zoznamu so všetkými kľúčmi slovníka spolu s hodnotami.
- “ tlačiť () “ sa používa na vytlačenie názvu a veľkosti parametra zadaním „ veľkosť () ” metóda a parameter:
To je všetko! Zostavili sme najjednoduchší spôsob tlače počtu parametrov modelu v PyTorch.
Záver
V PyTorch je „ nn.Modul ” sa používa na definovanie modelov, ktoré zahŕňajú všetky operácie a vrstvy tvoriace model. ' nn.Modul „trieda má „ parametre() ” metóda, ktorá sa používa na zobrazenie počtu parametrov modelu v modeli PyTorch. Tento zápis demonštroval metódu tlače počtu parametrov modelu v PyTorch.