[PyTorch] CIFAR-10 데이터셋 써보기
학습 위한 장치 얻기GPU 또는 MPS 같은 하드웨어 가속기에서 모델 학습`torch.cuda` 또는 `torch.backends.mps` 사용 가능한지 확인하고, 그렇지 않으면 CPU 사용if torch.cuda.is_available(): DEVICE = torch.device('cuda') elif torch.backends.mps.is_available(): DEVICE = torch.device('mps') else: DEVICE = torch.device('cpu') print('PyTorch version: ', torch.__version__, ', Device: ', DEVICE)transform = transforms.Compose( [transforms.ToTensor(), trans..
PROGRAMMING/AI
2023. 8. 7. 11:17