Internal change

PiperOrigin-RevId: 397179381

Internal change
PiperOrigin-RevId: 397179381
45fd9249 · Abdullah Rashwan · A. Unique TensorFlower · 1c79ece9 · 45fd9249
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

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

未找到文件。
--- 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)