See blogi kirjeldab:
- Millised on PyTorchi parameetrid?
- Miks peavad kasutajad mudeli parameetreid kontrollima?
- Kuidas PyTorchis mudeliparameetrite arvu kuvada?
Millised on PyTorchi parameetrid?
PyTorchis on ' nn.Moodul ” klassi kasutatakse mudelite määratlemiseks. See sisaldab kõiki toiminguid ja kihte, millest mudel koosneb. Iga kiht sisaldab parameetreid. Põhimõtteliselt värskendatakse parameetreid treenimise ajal, et minimeerida viga mudeli tegelike väärtuste ja prognooside vahel.
Miks peavad kasutajad mudeli parameetreid kontrollima?
Mudelit treenides peavad kasutajad teadma oma mudeli parameetrite arvu, kuna see võtab palju mälu ja töötlemisvõimsust. Kui nad on kursis mudeli parameetrite arvuga, saavad nad hõlpsasti hinnata, kui palju mälumahtu on vaja ja kui palju aega kulub treenimiseks, mis aitab kasutajatel oma treeningprotsessi optimeerida ja vältida süsteemi tühjenemist. ruumi.
Kuidas PyTorchis mudeliparameetrite arvu kuvada?
' nn.Moodul ' klassis on ' parameetrid () ” meetod, mida kasutatakse PyTorchi mudeli mudeliparameetrite arvu vaatamiseks. Kõigi elementide hankimiseks ' number1() ” meetodit kasutatakse.
Eelnevalt käsitletud kontseptsiooni mõistmiseks vaatame esitatud koodi:
importida tõrvik. nn nagu nn
klass NNmudel ( nn. Moodul ) :
def __kuum__ ( ise ) :
Super ( NNmudel , ise ) . __kuum__ ( )
ise . fc1 = nn. Lineaarne ( 10 , viiskümmend )
ise . fc2 = nn. Lineaarne ( viiskümmend , 1 )
def edasi ( ise , i ) :
i = ise . fc1 ( i )
i = ise . fc2 ( i )
tagasi i
minu_mudel = NNmudel ( )
t_params = summa ( lk. anna nimi ( ) jaoks lk sisse minu_mudel. parameetrid ( ) )
printida ( f 'Parameetrite koguarv: {t_params}' )
Ülaltoodud koodis:
- Esiteks määratleme mudeli, millel on kaks lineaarset kihti.
- Seejärel genereerige mudeli eksemplar ja kasutage parameetrid () ” meetod kõigi parameetrite hankimiseks.
- Järgmisena rakendame generaatori avaldist kõigi parameetrite arvutamiseks, summeerides iga parameetri elementide arvu.
- Lõpuks helistage ' print() ” avaldus tulemusväärtuste kuvamiseks ekraanil:
Ülalkirjeldatud koodis oleme kuvanud ainult parameetrite koguarvu, kui soovite saada parameetri nime ja suurust, saate kasutada järgmisi koodiridu:
jaoks nimi , param sisse minu_mudel. olek_dikt ( ) . esemed ( ) :printida ( nimi , param. suurus ( ) )
Siin:
- “ state_dict() ” on Pythoni sõnastiku objekt, mida kasutatakse PyTorchi mudelite salvestamiseks ja laadimiseks.
- “ üksus () ” meetodit kasutatakse loendi tagastamiseks koos kõigi sõnastikuvõtmete ja väärtustega.
- “ print() ' lauset kasutatakse parameetri nime ja suuruse printimiseks, edastades ' suurus () ” meetod ja parameeter:
See on kõik! Oleme koostanud lihtsaima viisi PyTorchis mudeli parameetrite arvu printimiseks.
Järeldus
PyTorchis on ' nn.Moodul ” klassi kasutatakse mudelite määratlemiseks, mis sisaldavad kõiki mudeli moodustavaid toiminguid ja kihte. ' nn.Moodul ' klassis on ' parameetrid () ” meetod, mida kasutatakse PyTorchi mudeli mudeliparameetrite arvu vaatamiseks. See kirjutis näitas PyTorchis mudeliparameetrite arvu printimise meetodit.