more trainable layers

d472c6b0 · sjjsmuel · 78bac83b · d472c6b0
Commit d472c6b0 authored Jun 08, 2020 by sjjsmuel
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

network/Resnet50.py network/Resnet50.py +1 -1

No files found.
--- a/network/Resnet50.py
+++ b/network/Resnet50.py
@@ -43,7 +43,7 @@ class Resnet50(NetworkBase):

        model = CAMModel(inputs=[input_tensor], outputs=[out, base_model.layers[-1].output])

-        for layer in model.layers[154:]:
+        for layer in model.layers[86:]:
            layer.trainable = True

        return model