Search

Model parameter 개수 세기

Created
2023/10/05 02:33
Tags
Status
Done
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
복사