[Pytorch] VGG 구현 및 정리
VGG 구현 및 정리 모두의 딥러닝 시즌2 - Pytorch를 참고 했습니다. 모두의 딥러닝 시즌2 깃헙 import torch.nn as nn class VGG(nn.Module): def __init__(self, features, num_classes=1000, init_weights=True): super(VGG, self).__init__() self.features = features self.avgpool = nn.AdaptiveAvgPool2d(7) self.classifier = nn.Sequential( nn.Linear(512*7*7, 4096), nn.ReLU(True), nn.Dropout(), nn.Linear(4096, 4096), nn.ReLU(True), nn.Dropout..
더보기
