Notice
Recent Posts
Recent Comments
Link
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
1 | 2 | 3 | 4 | |||
5 | 6 | 7 | 8 | 9 | 10 | 11 |
12 | 13 | 14 | 15 | 16 | 17 | 18 |
19 | 20 | 21 | 22 | 23 | 24 | 25 |
26 | 27 | 28 | 29 | 30 | 31 |
Tags
- 명령어
- pip
- 프로그래머스
- Numpy
- LIST
- Visual Studio
- Linux
- paramiko
- JSON
- OpenCV
- Python
- ubuntu
- YOLO
- label
- C
- 오류
- pytorch
- VS Code
- 기타 연주
- mysql
- SSH
- pandas
- 핑거스타일
- Docker
- error
- C#
- windows forms
- Selenium
- 채보
- C++
Archives
- Today
- Total
목록DataLoader (1)
기계는 거짓말하지 않는다
Pytorch Dataset, DataLoader
PyTorch는 torch.utils.data.Dataset과 torch.utils.data.DataLoader의 두 가지 데이터셋 라이브러리를 제공하며 미리 준비된(pre-loaded) 데이터셋 뿐 아니라 가지고 있는 데이터를 사용할 수 있다. Dataset은 data와 label을 저장한다. DataLoader는 Dataset에 쉽게 접근할 수 있도록 iterable 객체로 만들어 주고 sampler, shuffle, batch_size 등 다양한 매개변수를 설정 할 수 있다. from torch.utils.data import Dataset from torch.utils.data import DataLoader class CustomDataset(Dataset): def __init__(self, ..
AI
2021. 9. 5. 15:06