Loading DATA in PyTorch
Loading data in PyTorch — PyTorch Tutorials 1.12.1+cu102 documentation
PyTorch 具有廣泛的神經(jīng)網(wǎng)絡(luò)構(gòu)建塊,具有簡單、直觀和穩(wěn)定的 API。PyTorch 包含用于為模型準(zhǔn)備和加載常見數(shù)據(jù)集的包。
介紹
PyTorch 數(shù)據(jù)加載實用程序的核心是 torch.utils.data.DataLoader 類。它表示一個可迭代數(shù)據(jù)集的 Python。PyTorch 中的庫提供了內(nèi)置的高質(zhì)量數(shù)據(jù)集,供您在 torch.utils.data.Dataset 中使用。這些數(shù)據(jù)集目前在以下可用:
torchvision
torchaudio
torchtext
還有更多內(nèi)容即將推出。使用 torchaudio.datasets.YESNO 的Yesno數(shù)據(jù)集,我們將演示如何有效且高效地將數(shù)據(jù)從PyTorch數(shù)據(jù)集加載到PyTorch DataLoader中。