[PyTorch] PyTorch 기초 사용법
https://greeksharifa.github.io/pytorch/2018/11/10/pytorch-usage-03-How-to-Use-PyTorch/
torch.utils.data.DataLoader
학습에 사용될 전체 데이터를 가지고있다가, Train 함수가 1개의 batch를 요청 시 batch size 만큼 데이터를 꺼내서 준다.
ex) MNIST 데이터 불러오기
transform = transforms.Compose( [transforms.Resize( (28, 28) ),
transforms.ToTensor()
]
)
data_loader = DataLoader(datasets.MNSIT('data/MNIST',
train=True,
download=True,
transform=transform )
batch_size = 64
shuffle=True
)
print(data_loader)
transform
변수에 저렇게transforms.Compose
를 통해서 이미지(?)를 어떻게 변형시킬 것인지를 순차적으로 한꺼번에 적어준다.
torch.utils.data.DataLoader
를 통해서 DataLoader를 만들어준다.
torchvision.datasets
를 통해서 먼저 데이터를 받아주는것도 잊지말자!
[+]
len(data_loader.dataset)
= 실제 데이터의 개수
[+]len(data_loader)
= 실제 데이터의 개수 / batch_size
ex) 내가 가진 데이터셋의 데이터개수가 64,000장이고 batch_size를 64로 설정했을때
len(data_loader)
의 반환값은? 10,000
"뭐 약간 batch size만큼의 batch data를 10,000번 정도 줄게" 이런느낌?
작성중...
Author And Source
이 문제에 관하여([PyTorch] PyTorch 기초 사용법), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://velog.io/@olxtar/PyTorch-사용법저자 귀속: 원작자 정보가 원작자 URL에 포함되어 있으며 저작권은 원작자 소유입니다.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)