부스트캠프 AI Tech/2주차 필수과제 (1) 썸네일형 리스트형 [필수과제1] Multi-Layer Perceptron *Problem MNIST 데이터셋으로 pytorch를 활용한 MLP구현 *Code Import Library numpy matplotlib torch torch.nn as nn torch.optim as optim torch.nn.functional as F ++device setup : gpu or cpu device = torch.device(if torch.cuda.is_available() else 'cpu') Dataset from torchvision import datasets, transforms MNIST datasets(Train data sets : 60000 , Test data sets : 10000) Data Iterator torch.utils.data.DataLoader를 .. 이전 1 다음
