超簡單!用pytorch訓練自己的數據集

在pytorch官網上的tutoral中,教程給了一個例子,訓練一個分類器,下載CIFAR10數據集,用一個簡單的CNNC網絡訓練模型。在我們下載的CIFAR10數據集中,已標籤的數據被打包並封裝在data文件夾裏。 我們可以在這個例子的基礎上,把自己的數據放到其模型下,並實現一個簡易的分類器。 首先是配置pytorch的各種環境,有很多教程的,他們講的又詳細又好,我就不多講啦。 先導入各種模塊
相關文章
相關標籤/搜索