NAVER AI Tech/Pytorch

5. pytorch dataset

코딩소비 2023. 11. 16. 11:19

DataLoader 클래스를 사용하여 pytorch model의 input으로 제공하게 된다.

- tensor로 변환 + batch dataset 생성이 주된 업무.

- generator 형식이므로 for 문을 통해서 하나씩 받아오면 된다. next(), iter() 함수와 함께 사용!

- 여러가지 parameter가 있는데 document를 한번 찾아보자.