Internal change

PiperOrigin-RevId: 397179381

Internal change
PiperOrigin-RevId: 397179381
c67d2c08 · Abdullah Rashwan · A. Unique TensorFlower · 42ad9d5e · c67d2c08
Commit c67d2c08 authored Sep 16, 2021 by Abdullah Rashwan Committed by A. Unique TensorFlower Sep 16, 2021
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

official/vision/beta/projects/simclr/modeling/simclr_model.py ...cial/vision/beta/projects/simclr/modeling/simclr_model.py +2 -1

No files found.
--- a/official/vision/beta/projects/simclr/modeling/simclr_model.py
+++ b/official/vision/beta/projects/simclr/modeling/simclr_model.py
@@ -86,7 +86,8 @@ class SimCLRModel(tf.keras.Model):
      features = inputs

    # Base network forward pass.
-    endpoints = self._backbone(features, training=training)
+    endpoints = self._backbone(
+        features, training=training and self._backbone_trainable)
    features = endpoints[max(endpoints.keys())]
    projection_inputs = layers.GlobalAveragePooling2D()(features)