PyTorch學習筆記之數據加載和預處理

PyTorch 通過 torch.utils.data 對一般常用的數據加載進行了封裝,可以很容易地實現多線程數據預讀和批量加載。並且 torchvision 已經預先實現了常用圖像數據集,包括前面使用過的 CIFAR-10, ImageNet, COCO, MNIST, LSUN 等數據集,可通過 torchvision.datasets 方便的調用。   Dataset Dataset 是一個
相關文章
相關標籤/搜索