간단한 CNN 모델 만들기
import torchimport torch.nn as nnimport torch.optim as optim # 배치크기 * 채널(1: 그레이스케일, 3: 컬러, 4:컬러+투명도) * 너비 * 높이inputs = torch.Tensor(1, 1, 28, 28)print(inputs.shape) torch.Size([1, 1, 28, 28]) # 첫번째 Conv2Dconv1 = nn.Conv2d(in_channels=1, out_channels=32, kernel_size=3, padding='same')out = conv1(inputs)print(out.shape) torch.Size([1, 32, 28, 28]) # 첫번째 MaxPool2Dpool = nn.MaxPool2d(kernel_size=2)o..
2024. 7. 17.