Step 1️⃣ 拿预训练模型当底座
ResNet-18(ImageNet 1000类上预训练)
底层:边缘/纹理
中层:部件/形状
输出层:1000类 ❌换掉
Step 2️⃣ 换输出层 → 目标任务(如热狗识别,2类)
底层参数保留权重(初始化)+ 新随机初始化的输出层
🎯 训练技巧:底层用小学习率(或冻结),新输出层用大学习率
Generated with
Tabbit.