[머신러닝과 딥러닝] 20. 간단한 CNN 모델 만들기 + MNIST 분류하기
실습 1. 필요모듈 임포트 import torch import torch.nn as nn import torch.optim as optim 2. 차원지정 inputs = torch.Tensor(1, 1, 28, 28) print(inputs) # 결과값 => torch.Size([1, 1, 28, 28]) ▶ 1, 1, 28, 28 => 배치크기 * 채널(1: 그레이스케일, 3:컬러) * 높이 * 너비 3. 레이어 만들기 ▷ Conv2D conv1 = nn.Conv2d(in_channels=1, out_channels=32, kernel_size=3, padding='same') out = conv1(inputs) print(out.shape) # 결과값 => torch.Size([1, 32, 28, 2..
2024.01.10