Peter

Results 24 comments of Peter

加载完模型 .cuda()放到GPU上吧,你这好像确实在用cpu算

这个错误表明你尝试将一个大小为 1 的数组重塑为形状 (224, 224),但是这两个尺寸不匹配。在 NumPy 或类似的库中,reshape 操作用于改变数组的形状,但要确保新形状与原始数组的元素数量相匹配。检查下图片吧