python学习打卡:DAY 50 预训练模型+CBAM模块
所以很容易的想到之前第一次使用resnet的预训练策略:先冻结预训练层,然后训练其他层。之前的其它是全连接层(分类头),现在其它层还包含了每一个残差块中的cbam注意力层。预训练模型的结构和权重是固定的,如果修改其中的模型结构,是否会大幅影响其性能。其次是训练的时候如何训练才可以更好的避免破坏原有的特征提取器的参数。可以很明显的想到,如果是resnet18+cbam模块,那么大多数地方的代码都是可以复用的,模型定义部分需要重写。ps:今日的代码训练时长较长,3080ti大概需要40min的训练时长。







