Machine learning
ImageNet VGG
PyTorch ImageNet VGG16 none_avg_pool #1번 self.classifier = nn.Sequential( nn.Linear(512 * 7 * 7, 4096), nn.ReLU(True), nn.Dropout(), nn.Linear(4096, 4096), nn.ReLU(True), nn.Dropout(), nn.Linear(4096, 1000), ) def forward(self, x): x = self.features(x) x = x.view(x.size(0), -1) x = self.classifier(x) return x Epoch : 1 / Loss : 6.9131 / Time : 13.3H #2번 self.classifier = nn.Linear(7*7*512, 1000)..
