5. pytorch dataset
2023. 11. 16. 11:19ㆍNAVER AI Tech/Pytorch
DataLoader 클래스를 사용하여 pytorch model의 input으로 제공하게 된다.
- tensor로 변환 + batch dataset 생성이 주된 업무.
- generator 형식이므로 for 문을 통해서 하나씩 받아오면 된다. next(), iter() 함수와 함께 사용!
- 여러가지 parameter가 있는데 document를 한번 찾아보자.
'NAVER AI Tech > Pytorch' 카테고리의 다른 글
7. pytorch monitoring tools (0) | 2023.11.17 |
---|---|
6. 모델 저장, 불러오기 (0) | 2023.11.17 |
4. AutoGrad & Optimizer (0) | 2023.11.16 |
3. Pytorch 프로젝트 구조 (0) | 2023.11.13 |
2. Pytorch 기본 코드! (0) | 2023.11.13 |