问题描述:我在官方数据集上对模型进行训练没有问题,但转换图像尺寸后出现上述提到的问题。
问题原因:torch版本问题,torch2.2.2cu121有可能会出现这个问题。
解决方法:升级或降级torch。
解决依据:https://github.com/pytorch/pytorch/issues/115940
问题描述:我在官方数据集上对模型进行训练没有问题,但转换图像尺寸后出现上述提到的问题。
问题原因:torch版本问题,torch2.2.2cu121有可能会出现这个问题。
解决方法:升级或降级torch。
解决依据:https://github.com/pytorch/pytorch/issues/115940

被折叠的 条评论
为什么被折叠?