Model parameter 개수 세기

Deep learning model의 크기를 계산할 때, parameter 개수를 세서 report 하기도 한다.

아래 코드로 모델의 parameter 개수를 센다.

model = load_model(args)

# count number of elements
numel = sum([p.numel() for p in model.parameters()])
print(f"Model with {numel} parameters")
Python
복사