雖然沒認真調參數啦
但沒想到會輸weighted cross entropy
然後kornia的dice預設多分類
裡面有個one hot encode
我的東西丟進去會出現size問題
明天再想要改dice還是改我的程式
感覺直接改dice應該也不難