Како одштампати број параметара модела у ПиТорцх-у

Kako Odstampati Broj Parametara Modela U Pitorch U



ПиТорцх је популаран оквир који се користи у дубоком учењу. Нуди више функција за генерисање сложених неуронских мрежа (НН). Корисници могу да обављају операције обуке модела са овим оквиром. Међутим, корисници морају да буду упознати са бројем параметара пре него што обуче модел.

Овај блог ће описати:

Који су параметри у ПиТорцх-у?

У ПиТорцх-у, „ нн.Модуле ” класа се користи за дефинисање модела. Укључује све операције и слојеве који чине модел. Сваки слој садржи скуп параметара. Параметри се у основи ажурирају током обуке како би се минимизирала грешка између стварних вредности модела и предвиђања.







Зашто корисници морају да провере параметре модела?

Док обучавају модел, корисници морају да знају о броју параметара свог модела јер је потребно много меморије и процесорске снаге. Ако су упознати са бројем параметара модела, могу лако да процене количину меморије која ће им бити потребна и колико ће времена бити потребно за обуку, што помаже корисницима да оптимизују свој процес обуке, као и да спрече систем да остане без простор.



Како приказати број параметара модела у ПиТорцх-у?

нн.Модуле ” класа има “ параметри() ” метод који се користи за преглед броја параметара модела у ПиТорцх моделу. Да бисте добили све елементе, „ број1() ” се користи метода.



Да бисмо разумели концепт о коме је раније било речи, погледајмо дати код:





увоз бакља. нн као нн

класа ННМодел ( нн. Модул ) :
деф __вруће__ ( селф ) :
супер ( ННМодел , селф ) . __вруће__ ( )
селф . фц1 = нн. Линеар ( 10 , педесет )
селф . фц2 = нн. Линеар ( педесет , 1 )

деф напред ( селф , и ) :
и = селф . фц1 ( и )
и = селф . фц2 ( и )
повратак и

мој_модел = ННМодел ( )
т_парамс = сум ( стр. дати име ( ) за стр ин мој_модел. параметрима ( ) )
принт ( ф „Укупан број параметара: {т_парамс}“ )

У горе наведеном коду:

  • Прво, дефинишемо модел који има два линеарна слоја.
  • Затим генеришите инстанцу модела и користите „ параметри() ” за преузимање свих параметара.
  • Затим примењујемо израз генератора за израчунавање свих параметара сумирањем броја елемената сваког параметра.
  • На крају, позовите „ принт() ” за приказ резултујућих вредности на екрану:



У горе описаном коду, приказали смо само укупан број параметара, ако желите да добијете назив и величину параметра, могу се користити следеће линије кода:

за име , парам ин мој_модел. стате_дицт ( ) . ставке ( ) :

принт ( име , парам. величина ( ) )

овде:

  • стате_дицт() ” је Питхон речник објекат који се користи за складиштење и учитавање модела из ПиТорцх-а.
  • итем() ” метода се користи за враћање листе са свим кључевима речника заједно са вредностима.
  • принт() ” израз се користи за штампање имена и величине параметра преношењем „ величина() ” метод и параметар:

То је све! Саставили смо најлакши начин за штампање броја параметара модела у ПиТорцх-у.

Закључак

У ПиТорцх-у, „ нн.Модуле ” класа се користи за дефинисање модела који укључују све операције и слојеве који чине модел. „ нн.Модуле ” класа има “ параметри() ” метод који се користи за преглед броја параметара модела у ПиТорцх моделу. Овај запис је демонстрирао метод за штампање броја параметара модела у ПиТорцх-у.