trainable layers

e13d1751 · sjjsmuel · 982351b1 · e13d1751
Commit e13d1751 authored Apr 15, 2020 by sjjsmuel
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

classifier/Resnet50.py classifier/Resnet50.py +3 -0

No files found.
--- a/classifier/Resnet50.py
+++ b/classifier/Resnet50.py
@@ -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