前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
您好,我在训练安全帽数据集时发生如下报错:RuntimeError: stack expects each tensor to be equal size, but got [3, 480, 480] at entry 0 and [3, 448, 448] at entry 1,没有调试成功,请问大佬有无修改方法?