dnnsdunca commited on
Commit
ecd60f5
·
verified ·
1 Parent(s): ed72098

create unfreeze

Browse files
Files changed (1) hide show
  1. unfreeze +9 -0
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)