Този блог ще опише:
- Какви са параметрите в PyTorch?
- Защо потребителите трябва да проверяват параметрите на модела?
- Как да покажа броя на параметрите на модела в PyTorch?
Какви са параметрите в PyTorch?
В PyTorch, „ пп.Модул ” клас се използва за дефиниране на моделите. Той включва всички операции и слоеве, които изграждат модела. Всеки слой съдържа набор от параметри. Параметрите основно се актуализират по време на обучение, за да се сведе до минимум грешката между действителните стойности на модела и прогнозите.
Защо потребителите трябва да проверяват параметрите на модела?
Докато обучават модела, потребителите трябва да знаят за броя на параметрите на техния модел, тъй като отнема много памет и процесорна мощност. Ако са запознати с броя на параметрите на модела, те могат лесно да оценят количеството памет, което ще бъде необходимо и колко време ще отнеме за обучение, което помага на потребителите да оптимизират процеса на обучение, както и да предотвратят изчерпването на системата пространство.
Как да покажа броя на параметрите на модела в PyTorch?
„ пп.Модул ' класът има ' параметри() ”, който се използва за преглед на броя параметри на модела в модела на PyTorch. За да получите всички елементи, „ num1() ” се използва методът.
За да разберем обсъжданата по-рано концепция, нека да разгледаме предоставения код:
импортиране факла. nn като nn
клас NNМодел ( nn. Модул ) :
деф __горещо__ ( себе си ) :
супер ( NNМодел , себе си ) . __горещо__ ( )
себе си . fc1 = nn. Линеен ( 10 , петдесет )
себе си . fc2 = nn. Линеен ( петдесет , 1 )
деф напред ( себе си , i ) :
i = себе си . fc1 ( i )
i = себе си . fc2 ( i )
връщане i
моят_модел = NNМодел ( )
t_params = сума ( стр. дай име ( ) за стр в моят_модел. параметри ( ) )
печат ( f „Общ брой параметри: {t_params}“ )
В горния код:
- Първо дефинираме модел, който има два линейни слоя.
- След това генерирайте екземпляра на модела и използвайте „ параметри() ” за извличане на всички параметри.
- След това прилагаме генераторния израз, за да изчислим всички параметри, като сумираме броя на елементите на всеки параметър.
- И накрая, обадете се на „ печат () ” за показване на получените стойности на екрана:
В гореописания код сме показали само общия брой параметри, ако искате да получите името и размера на параметъра, можете да използвате следните редове код:
за име , парам в моят_модел. state_dict ( ) . елементи ( ) :печат ( име , парам. размер ( ) )
Тук:
- “ state_dict() ” е обектът на речника на Python, който се използва за съхраняване и зареждане на модели от PyTorch.
- “ вещ() ” методът се използва за връщане на списъка с всички речникови ключове заедно със стойности.
- “ печат () ' се използва за отпечатване на името и размера на параметъра чрез предаване на ' размер () ” метод и параметър:
Това е всичко! Компилирахме най-лесния начин за отпечатване на броя на параметрите на модела в PyTorch.
Заключение
В PyTorch, „ пп.Модул ” класът се използва за дефиниране на модели, които включват всички операции и слоеве, съставляващи модела. „ пп.Модул ' класът има ' параметри() ”, който се използва за преглед на броя параметри на модела в модела на PyTorch. Това описание демонстрира метода за отпечатване на броя на параметрите на модела в PyTorch.