【无标题】

以为是自己cuda安装有问题反复布了好几次环境最终发现环境根本没问题！！！！！！

将pred打印出来，返回值为nan的tensor量

查看自己的显卡！！！！！！是否支持半精度！！！！！

我的显卡是NVIDIA MX550 ，原模型中img做了半精度转换，我的显卡不支持半精度，所以转换成float。