CV

CV copied to clipboard

SegNet模型保存

9

您好，我在model_path导入ep050.pth的文件，但是他出现错误：KeyError: 'module.encoder.encode1.0.weight' 同时， ep050.pth在训练集与测试集loss均达0.05，但是，我用它predict的时候，只有一种颜色输出，甚至用训练集里的照片也是这样的结果 想请教一下： 请问是模型保存的出现问题了吗？ 或者是torch.nn.DataParallel的问题吗？（我的cuda定义是True，但是我只有一张显卡；可是我cuda定义是False时是cpu训练，很慢）

hocken-li

[train epoch 0] loss: 0.339386, acc: 7.600000: 100%| [valid epoch 0] loss: 0.001, acc 1.000: 100%| [train epoch 1] loss: 0.001157, acc: 8.000000: 100% 这准确率是多少啊？100%？还是说没显示？

xiaobai-ing

博主 请问如果要选择其中之一作为backbone进行训练得到预训练权重文件，是不是只要把输出通道数和分类的代码改掉就可以了？

Pillowmm

作者您好，为什么分类预测结果一直为daisy

4

ZZHOO1