pytorch #parameter #파이토치파라미터수계산 #python1 [Python/Pytorch] 네트워크 파라미터 수 계산법 / #params 계산법 Python, Pytorch 네트워크 파라미터 수 계산법 / # params 계산법 PyTorch에서 네트워크 모델의 파라미터 개수를 세는 것은 비교적 간단하다. 1. 총 파라미터 수 계산하기 모델의 모든 파라미터(학습 가능한 파라미터와 학습 불가능한 파라미터 포함)의 개수를 계산한다. 2. 학습 가능한 파라미터 수 계산하기 모델의 학습 가능한 파라미터(즉, 가중치)의 개수만을 계산한다. 이를 위한 간단한 함수는 다음과 같다. import torch.nn as nn def count_parameters(model): total_params = sum(p.numel() for p in model.parameters()) trainable_params = sum(p.numel() for p in model... 2023. 11. 17. 이전 1 다음
