deeplearning #딥러닝 #cnn #mnist1 CNN(합성곱 신경망) ② [MNIST] # 랜덤시드 고정하기 import random import torch import numpy as np import os seed = 42 random.seed(seed) np.random.seed(seed) torch.manual_seed(seed) if torch.cuda.is_available(): torch.cuda.manual_seed(seed) torch.cuda.manual_seed_all(seed) torch.backends.cudnn.benchmark = False torch.backends.cudnn.deterministic = True 1. Load MNIST Dataset # MNIST 데이터 셋 불러오기 from torchvision.datasets import MNIST down.. 2024. 4. 14. 이전 1 다음
