虽然没认真调参数啦
但没想到会输weighted cross entropy
然后kornia的dice默认多分类
里面有个one hot encode
我的东西丢进去会出现size问题
明天再想要改dice还是改我的程式
感觉直接改dice应该也不难