训练问题

[yangmenda]

大佬您好，我在训练自己数据集的时候遇到了一个问题，我设置num_classes为2（只需分割一个种类），但是在载入预训练模型时，始终无法载入 Fail To Load Key: ['cls_conv.weight', 'cls_conv.bias'] ……Fail To Load Key num: 2 这两个键，请问一下有没有上面解决办法？

[bubbliiiing]

你不看提示么

[NIU12377]

大佬您好，我在训练自己数据集的时候遇到了一个问题，我设置num_classes为2（只需分割一个种类），但是在载入预训练模型时，始终无法载入 Fail To Load Key： ['cls_conv.weight'， 'cls_conv.bias'] ......Fail To Load Key num： 2 这两个键，请问一下有没有上面解决办法？

你好，解决了吗