목록DataLoader (1)

HAZEL

[ PyTorch 02. ] Data Preprocess , DataLoader, 데이터 시각화

import os from glob import glob import torch from torchvision import datasets, transforms # dataset 예제 변환, transform 예제 변환을 줌 1. Data Loader 부르기 # batch 사이즈를 데이터 로드에 직접 넣어줌 batch_size = 32 test_batch_size =32 # train 용도 이므로 True , 로컬에 데이터가 없으면 download 받을 것이므로 True # 데이터를 변경시켜줄것이므로, 아래처럼 처리 train_loader = torch.utils.data.DataLoader( datasets.MNIST('dataset/', train = True , download= True, tran..
DATA ANALYSIS/ML & DL 2021. 4. 30. 18:28
이전 Prev 1 Next 다음