MNIST pytorch 썸네일형 리스트형 [Pytorch] MNIST 간단한 CNN 구현 및 정리 MNIST 간단한 CNN 구현 및 정리 모두의 딥러닝 시즌2 - Pytorch를 참고 했습니다. 모두의 딥러닝 시즌2 깃헙 import torch import torchvision.datasets as dsets import torchvision.transforms as transforms import torch.nn.init pytorch import device = 'cuda' if torch.cuda.is_available() else 'cpu' torch.manual_seed(777) if device == 'cuda': torch.cuda.manual_seed_all(123) device 설정(GPU or CPU) learning_rate = 0.001 training_epochs = 15 ba.. 더보기 이전 1 다음
