목록Pytorch pruning (1)

soyeonland

Pruning Tutorial

1. model 생성 import torch from torch import nn import torch.nn.utils.prune as prune import torch.nn.functional as F device = torch.device('cuda' if torch.cuda.is_available() else "cpu") class LeNet(nn.Module): def __init__(self): super(LeNet, self).__init__() self.conv1 = nn.Conv2d(1, 6, kernel_size=3) self.conv2 = nn.Conv2d(6, 16, kernel_size=3) self.fc1 = nn.Linear(16 * 5 *5, 120) self.fc2 = nn..
Study/Code Review 2020. 3. 22. 13:37
이전 Prev 1 Next 다음