Šajā emuārā tiks aprakstīts:
- Kādi ir PyTorch parametri?
- Kāpēc lietotājiem ir jāpārbauda modeļa parametri?
- Kā PyTorch parādīt modeļa parametru skaitu?
Kādi ir PyTorch parametri?
Programmā PyTorch “ nn.Modulis ” klase tiek izmantota modeļu definēšanai. Tas ietver visas darbības un slāņus, kas veido modeli. Katrs slānis satur noteiktu parametru kopu. Parametri pamatā tiek atjaunināti apmācības laikā, lai samazinātu kļūdu starp modeļa faktiskajām vērtībām un prognozēm.
Kāpēc lietotājiem ir jāpārbauda modeļa parametri?
Apmācot modeli, lietotājiem ir jāzina sava modeļa parametru skaits, jo tas aizņem daudz atmiņas un apstrādes jaudas. Ja viņi zina modeļa parametru skaitu, viņi var viegli novērtēt nepieciešamo atmiņas apjomu un laiku, kas būs nepieciešams apmācībai, kas palīdz lietotājiem optimizēt apmācības procesu, kā arī novērst sistēmas izsīkumu. telpa.
Kā PyTorch parādīt modeļa parametru skaitu?
' nn.Modulis 'klasē ir ' parametri () ” metode, kas tiek izmantota, lai skatītu modeļa parametru skaitu PyTorch modelī. Lai iegūtu visus elementus, ' num1() ” tiek izmantota metode.
Lai saprastu iepriekš apspriesto koncepciju, apskatīsim sniegto kodu:
imports lāpa. nn kā nn
klasē NNModelis ( nn. Modulis ) :
def __karsts__ ( sevi ) :
super ( NNModelis , sevi ) . __karsts__ ( )
sevi . fc1 = nn. Lineārs ( 10 , piecdesmit )
sevi . fc2 = nn. Lineārs ( piecdesmit , 1 )
def uz priekšu ( sevi , i ) :
i = sevi . fc1 ( i )
i = sevi . fc2 ( i )
atgriezties i
mans_modelis = NNModelis ( )
t_params = summa ( lpp. dot vārdu ( ) priekš lpp iekšā mans_modelis. parametrus ( ) )
drukāt ( f 'Kopējais parametru skaits: {t_params}' )
Iepriekš norādītajā kodā:
- Pirmkārt, mēs definējam modeli, kuram ir divi lineāri slāņi.
- Pēc tam ģenerējiet modeļa gadījumu un izmantojiet ' parametri () ” metodi, lai izgūtu visus parametrus.
- Tālāk mēs izmantojam ģeneratora izteiksmi, lai aprēķinātu visus parametrus, summējot katra parametra elementu skaitu.
- Visbeidzot, zvaniet uz ' drukāt () ” paziņojums, lai ekrānā parādītu iegūtās vērtības:
Iepriekš aprakstītajā kodā esam parādījuši tikai kopējo parametru skaitu, ja vēlaties iegūt parametra nosaukumu un izmēru, var izmantot šādas koda rindas:
priekš nosaukums , param iekšā mans_modelis. valsts_dikts ( ) . preces ( ) :drukāt ( nosaukums , param. Izmērs ( ) )
Šeit:
- ' state_dict() ” ir Python vārdnīcas objekts, kas tiek izmantots modeļu glabāšanai un ielādei no PyTorch.
- ' lieta() ” metode tiek izmantota, lai atgrieztu sarakstu ar visiem vārdnīcas taustiņiem kopā ar vērtībām.
- ' drukāt () ' paziņojums tiek izmantots, lai drukātu parametra nosaukumu un izmēru, nododot ' Izmērs() ” metode un parametrs:
Tas ir viss! Mēs esam apkopojuši vienkāršāko veidu, kā PyTorch izdrukāt modeļa parametru skaitu.
Secinājums
Programmā PyTorch “ nn.Modulis ” klase tiek izmantota, lai definētu modeļus, kas ietver visas operācijas un slāņus, kas veido modeli. ' nn.Modulis 'klasē ir ' parametri () ” metode, kas tiek izmantota, lai skatītu modeļa parametru skaitu PyTorch modelī. Šis raksts demonstrēja metodi modeļa parametru skaita drukāšanai PyTorch.