torchsummary1 [Pytorch] network parameter 갯수 확인 1. print(sum(p.numel() for p in model.parameters() if p.requires_grad)) 예를들어 다양한 모델에 대하여 from torchvision.models import * if __name__ == '__main__': model = vgg11() print("vgg11 : ", sum(p.numel() for p in model.parameters() if p.requires_grad)) model = vgg13() print("vgg13 : ", sum(p.numel() for p in model.parameters() if p.requires_grad)) model = vgg16() print("vgg16 : ", sum(p.numel() for p i.. 2023. 2. 1. 이전 1 다음
