Pytorch劃分數據集的方法

以前用過sklearn提供的劃分數據集的函數，以爲超級方便。可是在使用TensorFlow和Pytorch的時候一直找不到相似的功能，以前搜索的關鍵字都是「pytorch split dataset」之類的，可是搜出來仍是沒有我想要的。結果今天見鬼了忽然看見了這麼一個函數torch.utils.data.Subset。個人天，爲何超級開心hhhh。終於不用每次都手動劃分數據集了。html torc

>>阅读原文<<