ddroiddesignmoe / unfreeze
dnnsdunca's picture
create unfreeze
ecd60f5 verified
raw
history blame contribute delete
323 Bytes
# Unfreeze some layers
for layer in base_model.layers[-4:]:
layer.trainable = True
# Recompile the model (necessary after modifying layer.trainable)
model.compile(optimizer='adam', loss='binary_crossentropy', metrics=['accuracy'])
# Continue training
model.fit(train_data, epochs=10, validation_data=validation_data)