Ta blog bo opisal:
- Kakšni so parametri v PyTorchu?
- Zakaj morajo uporabniki preveriti parametre modela?
- Kako prikazati število parametrov modela v PyTorchu?
Kakšni so parametri v PyTorchu?
V PyTorchu je » nn.Modul ” razred se uporablja za definiranje modelov. Vključuje vse operacije in plasti, ki sestavljajo model. Vsaka plast vsebuje nabor parametrov. Parametri se v bistvu posodabljajo med usposabljanjem, da se zmanjša napaka med dejanskimi vrednostmi modela in napovedmi.
Zakaj morajo uporabniki preveriti parametre modela?
Med usposabljanjem modela morajo uporabniki poznati število parametrov svojega modela, ker zahteva veliko pomnilnika in procesorske moči. Če so seznanjeni s številom parametrov modela, lahko zlahka ocenijo količino pomnilnika, ki bo potreben, in koliko časa bo trajalo za vadbo, kar uporabnikom pomaga optimizirati njihov proces vadbe in prepreči, da bi sistemu zmanjkalo prostora.
Kako prikazati število parametrov modela v PyTorchu?
' nn.Modul ' razred ima ' parametri() ”, ki se uporablja za ogled števila parametrov modela v modelu PyTorch. Če želite pridobiti vse elemente, » num1() ” uporabljena metoda.
Da bi razumeli prej obravnavani koncept, si poglejmo priloženo kodo:
uvoz svetilka. nn kot nn
razred NNModel ( nn. Modul ) :
def __vroče__ ( sebe ) :
super ( NNModel , sebe ) . __vroče__ ( )
sebe . fc1 = nn. Linearno ( 10 , petdeset )
sebe . fc2 = nn. Linearno ( petdeset , 1 )
def naprej ( sebe , jaz ) :
jaz = sebe . fc1 ( jaz )
jaz = sebe . fc2 ( jaz )
vrnitev jaz
moj_model = NNModel ( )
t_params = vsota ( str. daj ime ( ) za str v moj_model. parametri ( ) )
tiskanje ( f 'Skupno število parametrov: {t_params}' )
V zgoraj navedeni kodi:
- Najprej definiramo model, ki ima dve linearni plasti.
- Nato ustvarite primerek modela in uporabite » parametri() ” za pridobitev vseh parametrov.
- Nato uporabimo generatorski izraz za izračun vseh parametrov tako, da seštejemo število elementov vsakega parametra.
- Na koncu pokličite » natisni() ” za prikaz rezultatov vrednosti na zaslonu:
V zgoraj opisani kodi smo prikazali samo skupno število parametrov, če želite pridobiti ime in velikost parametra, lahko uporabite naslednje vrstice kode:
za ime , param v moj_model. state_dict ( ) . predmete ( ) :tiskanje ( ime , param. velikost ( ) )
Tukaj:
- “ state_dict() ” je objekt slovarja Python, ki se uporablja za shranjevanje in nalaganje modelov iz PyTorcha.
- “ predmet() ” je uporabljena za vrnitev seznama z vsemi slovarskimi ključi skupaj z vrednostmi.
- “ natisni() ' se uporablja za tiskanje imena in velikosti parametra s posredovanjem ' velikost () ” metoda in parameter:
To je vse! Zbrali smo najlažji način za tiskanje števila parametrov modela v PyTorchu.
Zaključek
V PyTorchu je » nn.Modul ” se uporablja za definiranje modelov, ki vključujejo vse operacije in plasti, ki sestavljajo model. ' nn.Modul ' razred ima ' parametri() ”, ki se uporablja za ogled števila parametrov modela v modelu PyTorch. Ta zapis je pokazal metodo za tiskanje števila parametrov modela v PyTorchu.