pytorch实践(更改resnet50全连接层,重新训练,用于二分类)
背景交代(很杂乱,可跳过):最开始的目的是完成和SAN论文里一样,用把四种风格图片当做四种标签,在原来的权重基础上,整体重新训练resnet50两轮(原文中是resnet101),下图为resnet网络结构改变了最后的全连接层,从头开始重新训练,结果不太对,训练日志部分如下,一直不是等于100就是0Training dir : D:\postgraduate\major_rela...