Ovaj blog će opisati:
- Koji su parametri u PyTorchu?
- Zašto korisnici trebaju provjeriti parametre modela?
- Kako prikazati broj parametara modela u PyTorchu?
Koji su parametri u PyTorchu?
U PyTorchu, ' nn.Modul ” klasa se koristi za definiranje modela. Uključuje sve operacije i slojeve koji čine model. Svaki sloj sadrži skup parametara. Parametri se u osnovi ažuriraju tijekom obuke kako bi se smanjila pogreška između stvarnih vrijednosti modela i predviđanja.
Zašto korisnici trebaju provjeriti parametre modela?
Tijekom uvježbavanja modela, korisnici moraju znati o broju parametara svog modela jer je potrebno puno memorije i procesorske snage. Ako su upoznati s brojem parametara modela, lako mogu procijeniti količinu memorije koja će biti potrebna i koliko će vremena biti potrebno za treniranje, što pomaže korisnicima da optimiziraju svoj proces treniranja, kao i da spriječi da sustav ostane bez snage prostor.
Kako prikazati broj parametara modela u PyTorchu?
' nn.Modul ' klasa ima ' parametri() ” metoda koja se koristi za pregled broja parametara modela u PyTorch modelu. Da biste dobili sve elemente, ' num1() ” koristi se metoda.
Da bismo razumjeli prethodno razmatrani koncept, pogledajmo navedeni kod:
uvoz baklja. nn kao nn
razreda NNModel ( nn. Modul ) :
def __vruće__ ( sebe ) :
super ( NNModel , sebe ) . __vruće__ ( )
sebe . fc1 = nn. Linearno ( 10 , pedeset )
sebe . fc2 = nn. Linearno ( pedeset , 1 )
def naprijed ( sebe , ja ) :
ja = sebe . fc1 ( ja )
ja = sebe . fc2 ( ja )
povratak ja
moj_model = NNModel ( )
t_params = iznos ( str. daj ime ( ) za str u moj_model. parametri ( ) )
ispisati ( f 'Ukupan broj parametara: {t_params}' )
U gore navedenom kodu:
- Prvo definiramo model koji ima dva linearna sloja.
- Zatim generirajte instancu modela i upotrijebite ' parametri() ” za dohvaćanje svih parametara.
- Zatim primjenjujemo izraz generatora za izračunavanje svih parametara zbrajanjem broja elemenata svakog parametra.
- Na kraju, nazovite ' ispis() ” za prikaz rezultantnih vrijednosti na zaslonu:
U gore opisanom kodu prikazali smo samo ukupan broj parametara, ako želite dobiti naziv i veličinu parametra, možete koristiti sljedeće retke koda:
za Ime , param u moj_model. državni_dikt ( ) . stavke ( ) :ispisati ( Ime , param. veličina ( ) )
Ovdje:
- “ state_dict() ” je objekt Python rječnika koji se koristi za pohranu i učitavanje modela iz PyTorcha.
- “ artikal() ” metoda se koristi za vraćanje popisa sa svim ključevima rječnika zajedno s vrijednostima.
- “ ispis() ' naredba se koristi za ispis naziva i veličine parametra prosljeđivanjem ' veličina() ” metoda i parametar:
To je sve! Sastavili smo najlakši način ispisa broja parametara modela u PyTorchu.
Zaključak
U PyTorchu, ' nn.Modul ” klasa se koristi za definiranje modela koji uključuju sve operacije i slojeve koji čine model. ' nn.Modul ' klasa ima ' parametri() ” metoda koja se koristi za pregled broja parametara modela u PyTorch modelu. Ovaj zapis demonstrira metodu ispisa broja parametara modela u PyTorchu.