是在这一步遇到的：di = net(batch["image"].to(device))这两个问题我调试了1天半：结果发现！都是cuda的问题，# 确保所有输入都在指定设备上。