728x90 torch params1 torch model parameter(params) 개수 구하기 from torchvision import models def get_n_params(model): pp = 0 for p in list(model.parameters()): nn = 1 for s in list(p.size()): nn = nn * s pp += nn return pp resnet50_pretrained = models.resnet50(pretrained=True) parms_num = get_n_params(resnet50_pretrained) # 25557032 = 25.55M 2023. 7. 16. 이전 1 다음 728x90