Tento blog bude popisovat:
- Jaké jsou parametry v PyTorch?
- Proč uživatelé potřebují kontrolovat parametry modelu?
- Jak zobrazit počet parametrů modelu v PyTorch?
Jaké jsou parametry v PyTorch?
V PyTorch je „ nn.Modul ” třída se používá pro definování modelů. Zahrnuje všechny operace a vrstvy, které tvoří model. Každá vrstva obsahuje sadu parametrů. Parametry se v podstatě aktualizují během tréninku, aby se minimalizovala chyba mezi skutečnými hodnotami modelu a předpovědí.
Proč uživatelé potřebují kontrolovat parametry modelu?
Při trénování modelu uživatelé potřebují vědět o počtu parametrů svého modelu, protože vyžaduje hodně paměti a výpočetního výkonu. Pokud jsou obeznámeni s množstvím parametrů modelu, mohou snadno vyhodnotit množství paměti, které bude potřeba a kolik času zabere trénování, což pomáhá uživatelům optimalizovat jejich tréninkový proces a také zabránit vyčerpání systému. prostor.
Jak zobrazit počet parametrů modelu v PyTorch?
' nn.Modul třída má „ parametry() ” metoda, která se používá k zobrazení počtu parametrů modelu v modelu PyTorch. Chcete-li získat všechny prvky, „ číslo1() “ je použita metoda.
Abychom porozuměli dříve diskutovanému konceptu, podívejme se na poskytnutý kód:
import pochodeň. nn tak jako nn
třída NNModel ( nn. Modul ) :
def __horký__ ( já ) :
super ( NNModel , já ) . __horký__ ( )
já . fc1 = nn. Lineární ( 10 , padesáti )
já . fc2 = nn. Lineární ( padesáti , 1 )
def vpřed ( já , i ) :
i = já . fc1 ( i )
i = já . fc2 ( i )
vrátit se i
můj_model = NNModel ( )
t_params = součet ( p. Dej jméno ( ) pro p v můj_model. parametry ( ) )
tisk ( F 'Celkový počet parametrů: {t_params}' )
Ve výše uvedeném kódu:
- Nejprve definujeme model, který má dvě lineární vrstvy.
- Poté vygenerujte instanci modelu a použijte „ parametry() ” metoda pro načtení všech parametrů.
- Dále použijeme výraz generátoru k výpočtu všech parametrů sečtením počtu prvků každého parametru.
- Nakonec zavolejte „ tisk() ” pro zobrazení výsledných hodnot na obrazovce:
Ve výše popsaném kódu jsme zobrazili pouze celkový počet parametrů, pokud chcete získat název a velikost parametru, můžete použít následující řádky kódu:
pro název , param v můj_model. state_dict ( ) . položky ( ) :tisk ( název , param. velikost ( ) )
Tady:
- “ state_dict() ” je objekt slovníku Python, který se používá pro ukládání a načítání modelů z PyTorch.
- “ položka() ” metoda se používá k vrácení seznamu se všemi klíči slovníku spolu s hodnotami.
- “ tisk() 'příkaz se používá k vytištění názvu a velikosti parametru předáním ' velikost() ” metoda a parametr:
To je vše! Sestavili jsme nejjednodušší způsob, jak vytisknout počet parametrů modelu v PyTorch.
Závěr
V PyTorch je „ nn.Modul Třída ” se používá k definování modelů, které zahrnují všechny operace a vrstvy tvořící model. ' nn.Modul třída má „ parametry() ” metoda, která se používá k zobrazení počtu parametrů modelu v modelu PyTorch. Tento zápis demonstroval metodu tisku počtu parametrů modelu v PyTorch.