Šiame tinklaraštyje bus aprašyta:
- Kokie yra PyTorch parametrai?
- Kodėl vartotojai turi patikrinti modelio parametrus?
- Kaip „PyTorch“ parodyti modelio parametrų skaičių?
Kokie yra PyTorch parametrai?
„PyTorch“ nn.Modulis “ klasė naudojama modeliams apibrėžti. Tai apima visas operacijas ir sluoksnius, kurie sudaro modelį. Kiekviename sluoksnyje yra parametrų rinkinys. Parametrai iš esmės atnaujinami treniruojantis, kad būtų sumažinta paklaida tarp modelio faktinių verčių ir prognozių.
Kodėl vartotojai turi patikrinti modelio parametrus?
Treniruodami modelį vartotojai turi žinoti savo modelio parametrų skaičių, nes tai užima daug atminties ir apdorojimo galios. Jei jie yra susipažinę su modelio parametrų skaičiumi, jie gali lengvai įvertinti, kiek atminties reikės ir kiek laiko užtruks treniruotis, o tai padeda vartotojams optimizuoti mokymo procesą ir neleisti sistemai išsikrauti. erdvė.
Kaip „PyTorch“ parodyti modelio parametrų skaičių?
„ nn.Modulis “ klasė turi “ parametrai () “ metodas, naudojamas norint peržiūrėti modelio parametrų skaičių PyTorch modelyje. Norėdami gauti visus elementus, „ skaičius1 () “ naudojamas metodas.
Norėdami suprasti anksčiau aptartą koncepciją, pažvelkime į pateiktą kodą:
importuoti deglas. nn kaip nn
klasė NNModelis ( nn. Modulis ) :
def __karšta__ ( savarankiškai ) :
super ( NNModelis , savarankiškai ) . __karšta__ ( )
savarankiškai . fc1 = nn. Linijinis ( 10 , penkiasdešimt )
savarankiškai . fc2 = nn. Linijinis ( penkiasdešimt , 1 )
def Persiųsti ( savarankiškai , i ) :
i = savarankiškai . fc1 ( i )
i = savarankiškai . fc2 ( i )
grąžinti i
mano_modelis = NNModelis ( )
t_params = suma ( p. duok vardą ( ) dėl p in mano_modelis. parametrus ( ) )
spausdinti ( f 'Bendras parametrų skaičius: {t_params}' )
Aukščiau nurodytame kode:
- Pirmiausia apibrėžiame modelį, kuris turi du linijinius sluoksnius.
- Tada sugeneruokite modelio egzempliorių ir naudokite „ parametrai () “ būdas gauti visus parametrus.
- Toliau taikome generatoriaus išraišką, kad apskaičiuotume visus parametrus, susumavus kiekvieno parametro elementų skaičių.
- Galiausiai paskambinkite „ spausdinti () “ teiginys, kad ekrane būtų rodomos gautos reikšmės:
Aukščiau aprašytame kode parodėme tik bendrą parametrų skaičių, jei norite gauti parametro pavadinimą ir dydį, galite naudoti šias kodo eilutes:
dėl vardas , param in mano_modelis. valstybės_diktas ( ) . daiktų ( ) :spausdinti ( vardas , param. dydis ( ) )
Čia:
- “ state_dict() “ yra Python žodyno objektas, naudojamas modeliams iš PyTorch saugoti ir įkelti.
- “ prekė () “ metodas naudojamas norint grąžinti sąrašą su visais žodyno raktais ir reikšmėmis.
- “ spausdinti () ' teiginys naudojamas parametro pavadinimui ir dydžiui spausdinti perduodant ' dydis () “ metodas ir parametras:
Tai viskas! Mes sudarėme lengviausią būdą atspausdinti modelio parametrų skaičių „PyTorch“.
Išvada
„PyTorch“ nn.Modulis “ klasė naudojama modeliams, apimantiems visas operacijas ir sluoksnius, sudarančius modelį, apibrėžti. „ nn.Modulis “ klasė turi “ parametrai () “ metodas, naudojamas norint peržiūrėti modelio parametrų skaičių PyTorch modelyje. Šis užrašas parodė modelio parametrų skaičiaus spausdinimo „PyTorch“ metodą.