pytorch 自定義dataset

參考鏈接：https://www.jb51.net/article/167899.htm 在訓練神經網絡時，需要向網絡中丟入數據，以供神經網絡來學習其中的一些特徵，但是對於同樣的框架，神經網絡如何做到訓練各種各樣的數據呢？ 那麼就需要數據按照一定的格式來組織了，即Dataset類，（以便使用已經定義好的特殊數據集接口來加載數據） 1.先來介紹一下pytorch中的數據處理模塊torch.utils

>>阅读原文<<