Tämä blogi kertoo:
- Mitkä ovat PyTorchin parametrit?
- Miksi käyttäjien täytyy tarkistaa mallin parametrit?
- Kuinka näyttää malliparametrien lukumäärä PyTorchissa?
Mitkä ovat PyTorchin parametrit?
PyTorchissa ' nn.Moduuli ” -luokkaa käytetään mallien määrittelyyn. Se sisältää kaikki toiminnot ja tasot, jotka muodostavat mallin. Jokainen kerros sisältää joukon parametreja. Parametrit päivitetään periaatteessa koulutuksen aikana mallin todellisten arvojen ja ennusteiden välisen virheen minimoimiseksi.
Miksi käyttäjien täytyy tarkistaa mallin parametrit?
Harjoittaessaan mallia käyttäjien on tiedettävä mallinsa parametrien määrä, koska se vie paljon muistia ja prosessointitehoa. Jos he tuntevat mallin parametrien lukumäärän, he voivat helposti arvioida, kuinka paljon muistia tarvitaan ja kuinka paljon aikaa harjoitteluun kuluu, mikä auttaa käyttäjiä optimoimaan harjoitusprosessinsa ja estämään järjestelmän loppumisen. tilaa.
Kuinka näyttää malliparametrien lukumäärä PyTorchissa?
' nn.Moduuli 'luokassa on' parametrit () ” -menetelmä, jota käytetään PyTorch-mallin malliparametrien määrän tarkasteluun. Saadaksesi kaikki elementit, ' numero1() ”menetelmää käytetään.
Ymmärtääksesi aiemmin käsiteltyä käsitettä, katsotaanpa annettua koodia:
tuonti taskulamppu. nn kuten nn
luokkaa NNMalli ( nn. Moduuli ) :
def __kuuma__ ( itse ) :
super ( NNMalli , itse ) . __kuuma__ ( )
itse . fc1 = nn. Lineaarinen ( 10 , viisikymmentä )
itse . fc2 = nn. Lineaarinen ( viisikymmentä , 1 )
def eteenpäin ( itse , i ) :
i = itse . fc1 ( i )
i = itse . fc2 ( i )
palata i
minun_mallini = NNMalli ( )
t_params = summa ( s. anna nimi ( ) varten s sisään minun_mallini. parametrit ( ) )
Tulosta ( f 'Parametrien kokonaismäärä: {t_params}' )
Yllä mainitussa koodissa:
- Ensin määritetään malli, jossa on kaksi lineaarista kerrosta.
- Luo sitten mallin ilmentymä ja käytä ' parametrit () ' -menetelmää kaikkien parametrien hakemiseksi.
- Seuraavaksi käytämme generaattorilauseketta kaikkien parametrien laskemiseen summaamalla kunkin parametrin elementtien lukumäärä.
- Lopuksi soita ' Tulosta() ”-lause näyttää tuloksena olevat arvot näytöllä:
Yllä kuvatussa koodissa olemme näyttäneet vain parametrien kokonaismäärän, jos haluat saada parametrin nimen ja koon, voit käyttää seuraavia koodirivejä:
varten nimi , param sisään minun_mallini. state_dict ( ) . kohteita ( ) :Tulosta ( nimi , param. koko ( ) )
Tässä:
- ' state_dict() ” on Python-sanakirjaobjekti, jota käytetään mallien tallentamiseen ja lataamiseen PyTorchista.
- ' tuote() ” -menetelmää käytetään palauttamaan luettelo, jossa on kaikki sanakirjan avaimet arvoineen.
- ' Tulosta() ' -lausetta käytetään parametrin nimen ja koon tulostamiseen välittämällä ' koko() ” menetelmä ja parametri:
Siinä kaikki! Olemme koonneet helpoimman tavan tulostaa malliparametrien lukumäärä PyTorchissa.
Johtopäätös
PyTorchissa ' nn.Moduuli ” -luokkaa käytetään sellaisten mallien määrittämiseen, jotka sisältävät kaikki mallin muodostavat toiminnot ja tasot. ' nn.Moduuli 'luokassa on' parametrit () ” -menetelmä, jota käytetään PyTorch-mallin malliparametrien määrän tarkasteluun. Tämä kirjoitus osoitti menetelmän malliparametrien lukumäärän tulostamiseksi PyTorchissa.