如何允许最后一个resnet块可训练？

Question

我想使最后一个resnet块是可训练的，因为我怀疑从ImageNet学习的特征可能与我的图像分类问题没有直接关系

我设法在最后一个街区对VG16进行了培训，并想知道如何将同样的方法也应用到resnet中

model.get_layer('block5_conv1').trainable = True
model.get_layer('block5_conv2').trainable = True
model.get_layer('block5_conv3').trainable = True
model.get_layer('block5_pool').trainable = True

Answer 1

当然。一般来说，保持大多数层不变是个好主意，只需重新训练最后几层。例如，这就是冻结除最后一层以外的所有层的方法

for layer in model.layers[0:-1]:
  layer.trainable = False