인공지능/DeepLearning
2020. 12. 11.
[CNN] Alex Net - 구현 및 MNIST_Fashion 학습
AlexNet을 모듈로 표현해보기 위해 구현하였으며 이미지 또한 AlexNet에 맞춰 사용이 가능하도록 구현하였다. AlexNet은 2개의 GPU로 병렬 연산한다는게 포인트이지만, 그래픽카드가 하나인 관계로 병렬연산은 구현하지 못했다. AlexNet 구현 import torch import torch.nn as nn import torch.nn.functional as F from torchvision import transforms class AlexNet(nn.Module): def __init__(self): super(AlexNet, self).__init__() self.conv1 = nn.Conv2d(in_channels=1, out_channels=96, kernel_size=11, stri..
