[DL] CNN 모델만들기,MNLIST 분류
1. 간단한 CNN모델 만들기import torchimport torch.nn as nnimport torch.optim as optimCNN모델을 간단하게 만들기 위해서 필요한 모듈들을 임포트해주었씁니다. # 배치크기 * 채널 (1: 그레이스킬, 3: 트루컬러) *너비 * 높이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..
2024. 7. 28.