[Pytorch] 간단한 VGG16 코드 (수정중)
안녕하세요! 이번에는 VGG16 에 대해 공부한걸 작성해보겠습니다. 잘못된게 있다면 조언,충고 부탁합니다 :) import torch.nn as nn import torch x = torch.randn((1,3,256,256)).float() class VGG_A(nn.Module): def __init__(self, num_classes: int = 1000, init_weights: bool = True): super(VGG_A, self).__init__() self.convnet = nn.Sequential( # Input Channel (RGB: 3) nn.Conv2d(in_channels=3, out_channels=64, kernel_size=3, padding=1, stride=1), nn..
2022. 11. 9.