【Pytorch】問題解決:RuntimeError: stack expects each tensor to be equal size, but got [109] at entry 0 and [110] at entry 1

前言

這篇是我在使用 Pytorch 的時候碰到的問題,出現了以下的訊息:

RuntimeError: stack expects each tensor to be equal size, but got [109] at entry 0 and [110] at entry 1

解決方法

我碰到這個問題的時候,最後解決問題的方法是發現 label 的名稱錯誤!
原本的 label 應該是 2-75 的數值,
但因為程式寫錯,變成只有 109, 110 兩種數值,導致這個錯誤!!

遇到此問題時,請檢查每一筆資料的 label !!!

Licensed under CC BY-NC-SA 4.0