목록dataloader (1)

HealthyAI

PyTorch) Dataset & DataLoader

Dataset & DataLoader Deep Learning Process Data - collecting, cleaning, pre processing Dataset ToTensor(), CenterCrop() transforms init() call() Dataset init() len() getitem() DataLoader batch, shuffle Model Dataset class 데이터의 형태를 정의하는 클래스 데이터를 입력하는 방식의 표준화 Image, Text, Audio 등에 따른 다른 입력정의 import torch from torch.utils.data import Dataset class CustomDataset(Dataset): def __init__(self, text, la..
AI/PyTorch 2023. 3. 18. 14:14
이전 Prev 1 Next 다음