more trainable then before

e580ed83 · sjjsmuel · 4741732d · e580ed83
Commit e580ed83 authored Apr 11, 2020 by sjjsmuel
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

classifier/Resnet152.py classifier/Resnet152.py +2 -2

No files found.
--- a/classifier/Resnet152.py
+++ b/classifier/Resnet152.py
@@ -38,7 +38,7 @@ class Resnet152(NetworkBase):

        model = Model(base_model.input, out)

-        #for layer in model.layers[:-5]:
-        #    layer.trainable = False
+        for layer in model.layers[:-30]:
+            layer.trainable = False

        return model