파라미터 수1 torchsummary를 이용해서 모델 파라미터 수 파악하기 $pip install torchsummary 일단 torchsummary 라이브러리를 설치해준다. import torch from torchvision import models from torchsummary import summary device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') vgg = models.vgg16().to(device) summary(vgg, (3, 224, 224)) 이런식으로 모델을 담은 변수, input 값의 사이즈를 인자로 넘겨주면 ---------------------------------------------------------------- Layer (type) Output Shape Para.. 2021. 10. 11. 이전 1 다음