목록dnn (1)
David의 개발 이야기!
DNN을 활용한 MNIST 분류 모델 구현
DNN == Deep Neural Network 심층 신경망(Deep Neural Network, DNN)은 입력층(input layer)과 출력층(output layer) 사이에 여러 개의 은닉층(hidden layer)들로 이뤄진 인공신경망(Artificial Neural Network, ANN)이다. 이를 이용하여, 손글씨 숫자이미지를 분류하는 모델을 구현해보고자 한다. 1. 필요한 모듈 불러오기 import torch import torch.nn as nn import torch.optim as optim import torch.nn.functional as F from torchvision import datasets, transforms device = torch.device("cuda:0" ..
인공지능공부
2023. 8. 7. 22:31