output with shape [1, 448, 448] doesn't match the broadcast shape [3, 448, 448]跑的过程中出现错误

问题遇到的现象和发生背景

跑模型的过程中出现此错误，应该如何解决，为何模型一开始能跑，却在过程中发生错误？

用代码块功能插入代码，请勿粘贴截图

通道数不对，看下数据集里面是不是单通道灰度图和RGB混淆了