Commit e13d1751 authored by sjjsmuel's avatar sjjsmuel

trainable layers

parent 982351b1
......@@ -34,5 +34,8 @@ class Resnet50(NetworkBase):
out = Dense(self.NUM_CLASSES, activation='softmax', name='probs')(x)
model = Model(base_model.input, out)
for layer in model.layers[:-39]:
layer.trainable = False
return model
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment