報錯的原因是,不同的數據長度不同,無法組成一個batch tensor。 DataLoader 中有個參數 collate_fn ,專門用來把Dataset 類的返回值拼接成tensor,我們不設置的時候 ...
確定! 回上一頁