PyTorch在學術界和工業界的應用研究中都獲得了很多關注。它是一個具有很大靈活性的深度學習框架，使用了大量的實用工具和函數來加快工作速度。PyTorch的學習曲線并不是那么陡峭，但在其中實現高效和干凈的代碼可能會很棘手。在使用它超過2年之后，以下是我最喜歡的PyTorch功能，我希望我一開始學習它就知道。

1. DatasetFolder

當學習PyTorch時，人們首先要做的事情之一是實現自己的某種Dataset 。這是一個低級錯誤，沒有必要浪費時間寫這樣的東西。通常，數據集要么是數據列表(或者是numpy數組)，要么磁盤上的文件。所以，把數據在磁盤上組織好，要比寫一個自定義的Dataset來加載某種奇怪的格式更好。

分類器最常見的數據格式之一，是有一個帶有子文件夾的目錄，子文件夾表示類，子文件夾中的文件表示樣本，如下所示。

folder/class_0/file1.txt folder/class_0/file2.txt folder/class_0/...folder/class_1/file3.txt folder/class_1/file4.txtfolder/class_2/file5.txt folder/class_2/...

有一個內置的方式來加載這類數據集，不管你的數據是圖像，文本文件或其他什么，只要使用'DatasetFolder就可以了。令人驚訝的是，這個類是torchvision包的一部分，而不是核心PyTorch。這個類非常全面，你可以從文件夾中過濾文件，使用自定義代碼加載它們，并動態轉換原始文件。例子:

from torchvision.datasets im

總結

以上是生活随笔為你收集整理的PyTorch的13个必知必会知识点的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。