Spaces:
Runtime error
Runtime error
create unfreeze
Browse files
unfreeze
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Unfreeze some layers
|
| 2 |
+
for layer in base_model.layers[-4:]:
|
| 3 |
+
layer.trainable = True
|
| 4 |
+
|
| 5 |
+
# Recompile the model (necessary after modifying layer.trainable)
|
| 6 |
+
model.compile(optimizer='adam', loss='binary_crossentropy', metrics=['accuracy'])
|
| 7 |
+
|
| 8 |
+
# Continue training
|
| 9 |
+
model.fit(train_data, epochs=10, validation_data=validation_data)
|